tanyarai commited on
Commit
be1bd10
1 Parent(s): 6a3fb13

Create model_parsers.py

Browse files
Files changed (1) hide show
  1. model_parsers.py +17 -0
model_parsers.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from aiconfig import AIConfigRuntime, ModelParserRegistry, DalleImageGenerationParser
2
+
3
+
4
+ # Example of how users can register model parsers for use in the GradioWorkbook
5
+ # The implementation looks for a parsers_path (model_parsers.py by default) which
6
+ # should include a module with a register_model_parsers function.
7
+ # Here we are registering all the local HuggingFace model parsers as an example
8
+ def register_model_parsers() -> None:
9
+
10
+ dalle_image_generation_models = [
11
+ "dall-e-2",
12
+ "dall-e-3",
13
+ ]
14
+ for model in dalle_image_generation_models:
15
+ ModelParserRegistry.register_model_parser(
16
+ DalleImageGenerationParser(model)
17
+ )