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)
|