Spaces:
Runtime error
Runtime error
Update model_parsers.py
Browse files- model_parsers.py +7 -6
model_parsers.py
CHANGED
@@ -1,9 +1,10 @@
|
|
1 |
-
from
|
2 |
-
from aiconfig
|
3 |
|
4 |
|
5 |
-
# By default, gradio implementation will register all supported HF remote inference models
|
6 |
-
# For this example, we only want Dall-E, so clear all registered models and then add Dall-E
|
7 |
def register_model_parsers() -> None:
|
8 |
-
|
9 |
-
ModelParserRegistry.register_model_parser(
|
|
|
|
|
|
|
|
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")
|
7 |
+
ModelParserRegistry.register_model_parser(groq_mixtral)
|
8 |
+
|
9 |
+
groq_llama = GroqParser(model="llama2-70b-4096")
|
10 |
+
ModelParserRegistry.register_model_parser(groq_llama)
|