Adding CogVideo

model/model_registry.py

@@ -213,6 +213,13 @@ register_model_info(
     "https://arxiv.org/abs/2308.06571",
     "ModelScope is a a T2V synthesis model that evolves from a T2I synthesis model.",
 )
+
+register_model_info(
+    ["videogenhub_CogVideo_generation"],
+    "CogVideo",
+    "https://arxiv.org/abs/2205.15868",
+    "Text-to-Video Generation via Transformers",
+)
     
 
 models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation',
@@ -221,4 +228,4 @@ models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub
           'imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
           'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',
           "fal_AnimateDiffTurbo_text2video", "fal_AnimateDiff_text2video",
-          "videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation"]
+          "videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation", "videogenhub_CogVideo_generation"]

model/models/__init__.py

@@ -13,7 +13,7 @@ IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZe
 VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                            'fal_AnimateDiffTurbo_text2video',
                            'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
-                           'videogenhub_ModelScope_generation']
+                           'videogenhub_ModelScope_generation', 'videogenhub_CogVideo_generation']
 
 
 def load_pipeline(model_name):