Adding Allegro

model/model_registry.py

@@ -449,3 +449,13 @@ register_model_info(
     "text2video_generation"
 )
     
+register_model_info(
+    ["videogenhub_Allegro_text2video"],
+    "Allegro",
+    "https://github.com/rhymes-ai/Allegro",
+    "DiT based Video Generation Model",
+    "Apache 2.0",
+    "rhymes-ai",
+    "text2video_generation"
+)
+    

model/models/__init__.py

@@ -28,7 +28,8 @@ VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                            'fal_T2VTurbo_text2video',
                            'fal_StableVideoDiffusion_text2video',
                            'fal_CogVideoX-5B_text2video',
-                           'videogenhub_PyramidFlow_text2video']
+                           'videogenhub_PyramidFlow_text2video',
+                           'videogenhub_Allegro_text2video']
 
 MAP_NAMES_IMAGENHUB = {}
 MAP_NAMES_VIDEOGENHUB = {"CogVideoX-2B": "CogVideoX", "CogVideoX-5B": "CogVideoX5B"}