Spaces:
Running
Running
File size: 400 Bytes
5a93166 7c25085 cda0f85 5a93166 7c25085 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from aiconfig_extension_groq import GroqParser
from aiconfig import ModelParserRegistry
import dotenv
def register_model_parsers() -> None:
groq_mixtral = GroqParser(model="mixtral-8x7b-32768")
ModelParserRegistry.register_model_parser(groq_mixtral)
groq_llama = GroqParser(model="llama2-70b-4096")
ModelParserRegistry.register_model_parser(groq_llama)
dotenv.load_dotenv() |