update

model/model_registry.py

@@ -1,30 +1,6 @@
 from collections import namedtuple
 from typing import List
 
-IMAGE_GENERATION_MODELS = ['imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation', 'imagenhub_PixArtSigma_generation',
-                           'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation', 'imagenhub_HunyuanDiT_generation',
-                           'playground_PlayGroundV2.5_generation', 'imagenhub_Kolors_generation', 'imagenhub_SD3_generation',
-                           'fal_AuraFlow_text2image', 'fal_FLUX1schnell_text2image', 'fal_FLUX1dev_text2image']  # 'playground_PlayGroundV2_generation'
-IMAGE_EDITION_MODELS = ['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', '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_T2VTurbo_text2video',
-                           'fal_StableVideoDiffusion_text2video']
-MUSEUM_UNSUPPORTED_MODELS = ['videogenhub_OpenSoraPlan_generation']
-DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
-
-ALL_MODELS = IMAGE_GENERATION_MODELS + IMAGE_EDITION_MODELS + VIDEO_GENERATION_MODELS
-
-
 ModelInfo = namedtuple("ModelInfo", ["simple_name", "link", "description", "license", "organization", "type"])
 model_info = {}
 
@@ -454,5 +430,3 @@ register_model_info(
     "text2video_generation"
 )
     
-
-assert all([model in model_info for model in ALL_MODELS]), "Some models are not registered in model_info"