Update model/models/fal_api_models.py

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", "T2VTurbo": "t2v-turbo", "CogVideoX": "cogvideox-5b"}
+                      "AuraFlow": "aura-flow", "FLUX1schnell": "flux/schnell", "FLUX1dev": "flux/dev", "T2VTurbo": "t2v-turbo"}
 
 class FalModel():
     def __init__(self, model_name, model_type):
@@ -62,6 +62,8 @@ class FalModel():
                 fal_model_name = 'fast-animatediff/turbo/text-to-video'
             elif self.model_name == 'StableVideoDiffusion':
                 fal_model_name = 'fast-svd/text-to-video'
+            elif self.model_name == 'CogVideoX':
+                fal_model_name = 'cogvideox-5b'
             else:
                 raise NotImplementedError(f"text2video model of {self.model_name} in fal is not implemented yet")
             handler = fal_client.submit(