Adding CogVideoX5B precompute

model/fetch_museum_results/__init__.py

@@ -3,7 +3,7 @@ from .imagen_museum import fetch_indexes, fetch_indexes_no_csv
 import random
 
 ARENA_TO_IG_MUSEUM = {"LCM(v1.5/XL)":"LCM", "PlayGroundV2.5": "PlayGroundV2_5", "FLUX.1-schnell": "FLUX1schnell", "FLUX.1-dev": "FLUX1dev"}
-ARENA_TO_VG_MUSEUM = {"StableVideoDiffusion": "FastSVD", "CogVideoX-2B": "CogVideoX"}
+ARENA_TO_VG_MUSEUM = {"StableVideoDiffusion": "FastSVD", "CogVideoX-2B": "CogVideoX", "CogVideoX-5B": "CogVideoX5B"}
 
 def draw2_from_imagen_museum(task, model_name1, model_name2):
     task_name = TASK_DICT[task]

model/models/__init__.py

@@ -30,9 +30,9 @@ VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                            'fal_CogVideoX-5B_text2video']
 
 MAP_NAMES_IMAGENHUB = {}
-MAP_NAMES_VIDEOGENHUB = {"CogVideoX-2B": "CogVideoX"}
+MAP_NAMES_VIDEOGENHUB = {"CogVideoX-2B": "CogVideoX", "CogVideoX-5B": "CogVideoX5B"}
 
-MUSEUM_UNSUPPORTED_MODELS = ['fal_CogVideoX-5B_text2video']
+MUSEUM_UNSUPPORTED_MODELS = []
 DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
 
 ALL_MODELS = IMAGE_GENERATION_MODELS + IMAGE_EDITION_MODELS + VIDEO_GENERATION_MODELS