tanyarai commited on
Commit
5a93166
1 Parent(s): 95529cc

Update model_parsers.py

Browse files
Files changed (1) hide show
  1. model_parsers.py +7 -6
model_parsers.py CHANGED
@@ -1,9 +1,10 @@
1
- from aiconfig import AIConfigRuntime, ModelParserRegistry
2
- from aiconfig.default_parsers.dalle import DalleImageGenerationParser
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
- ModelParserRegistry.clear_registry()
9
- ModelParserRegistry.register_model_parser(DalleImageGenerationParser('dall-e-3'))
 
 
 
 
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)