Adding PyramidFlow

model/model_registry.py

@@ -420,6 +420,15 @@ register_model_info(
     "text2video_generation"
 )
 
+register_model_info(
+    ["videogenhub_PyramidFlow_text2video"],
+    "Pyramid Flow",
+    "https://pyramid-flow.github.io/",
+    "Pyramidal Flow Matching for Efficient Video Generative Modeling.",
+    "MIT LICENSE",
+    "Peking University",
+    "text2video_generation"
+)
 register_model_info(
     ["fal_CogVideoX-5B_text2video"],
     "CogVideoX-5B",

model/models/__init__.py

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