Spaces:
Running
Running
Update model_parsers.py
Browse files- model_parsers.py +3 -1
model_parsers.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from aiconfig_extension_groq import GroqParser
|
2 |
from aiconfig import ModelParserRegistry
|
3 |
-
|
4 |
|
5 |
def register_model_parsers() -> None:
|
6 |
groq_mixtral = GroqParser(model="mixtral-8x7b-32768")
|
@@ -8,3 +8,5 @@ def register_model_parsers() -> None:
|
|
8 |
|
9 |
groq_llama = GroqParser(model="llama2-70b-4096")
|
10 |
ModelParserRegistry.register_model_parser(groq_llama)
|
|
|
|
|
|
1 |
from aiconfig_extension_groq import GroqParser
|
2 |
from aiconfig import ModelParserRegistry
|
3 |
+
import dotenv
|
4 |
|
5 |
def register_model_parsers() -> None:
|
6 |
groq_mixtral = GroqParser(model="mixtral-8x7b-32768")
|
|
|
8 |
|
9 |
groq_llama = GroqParser(model="llama2-70b-4096")
|
10 |
ModelParserRegistry.register_model_parser(groq_llama)
|
11 |
+
|
12 |
+
dotenv.load_dotenv()
|