dropping some t2v models

model/model_registry.py

@@ -258,6 +258,7 @@ register_model_info(
     "AnimateDiff Turbo is a lightning version of AnimateDiff.",
 )
 
+"""
 register_model_info(
     ["videogenhub_LaVie_generation"],
     "LaVie",
@@ -265,6 +266,7 @@ register_model_info(
     "LaVie is a video generation model with cascaded latent diffusion models.",
 )
 
+
 register_model_info(
     ["videogenhub_VideoCrafter2_generation"],
     "VideoCrafter2",
@@ -285,7 +287,7 @@ register_model_info(
     "https://github.com/hpcaitech/Open-Sora",
     "A community-driven opensource implementation of Sora.",
 )
-
+"""
 register_model_info(
     ["videogenhub_OpenSora12_generation"],
     "OpenSora v1.2",
@@ -301,7 +303,7 @@ register_model_info(
 )
     
 register_model_info(
-    ["videogenhub_T2VTurbo_generation"],
+    ["fal_T2VTurbo_generation"],
     "T2V-Turbo",
     "https://github.com/Ji4chenLi/t2v-turbo",
     "Video Consistency Model with Mixed Reward Feedback.",

model/models/__init__.py

@@ -17,10 +17,14 @@ IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZe
                         'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition', 'imagenhub_UltraEdit_edition']
 VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                            'fal_AnimateDiffTurbo_text2video',
-                           'videogenhub_LaVie_generation', 
-                           'videogenhub_VideoCrafter2_generation',
-                           'videogenhub_ModelScope_generation', 'videogenhub_CogVideoX_generation', 'videogenhub_OpenSora12_generation',
-                           'videogenhub_OpenSora_generation', 'videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
+                           #'videogenhub_LaVie_generation', 
+                           #'videogenhub_VideoCrafter2_generation',
+                           #'videogenhub_ModelScope_generation', 
+                           'videogenhub_CogVideoX_generation', 'videogenhub_OpenSora12_generation',
+                           #'videogenhub_OpenSora_generation', 
+                           #'videogenhub_T2VTurbo_generation',
+                           'fal_T2VTurbo_text2video',
+                           'fal_StableVideoDiffusion_text2video']
 MUSEUM_UNSUPPORTED_MODELS = ['videogenhub_OpenSoraPlan_generation']
 DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
 

model/models/fal_api_models.py

@@ -7,7 +7,7 @@ import base64
 
 FAL_MODEl_NAME_MAP = {"SDXL": "fast-sdxl", "SDXLTurbo": "fast-turbo-diffusion", "SDXLLightning": "fast-lightning-sdxl",
                       "LCM(v1.5/XL)": "fast-lcm-diffusion", "PixArtSigma": "pixart-sigma", "StableCascade": "stable-cascade",
-                      "AuraFlow": "aura-flow", "FLUX1schnell": "flux/schnell", "FLUX1dev": "flux/dev"}
+                      "AuraFlow": "aura-flow", "FLUX1schnell": "flux/schnell", "FLUX1dev": "flux/dev", "T2VTurbo": "t2v-turbo"}
 
 class FalModel():
     def __init__(self, model_name, model_type):