Spaces:

TIGER-Lab
/

GenAI-Arena

Running on Zero

vinesmsuic commited on Aug 28

Commit

18c9c6f

•

1 Parent(s): dfa8b32

Adding CogVideoX fal

Files changed (1) hide show

model/models/fal_api_models.py CHANGED Viewed

@@ -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"}
 class FalModel():
     def __init__(self, model_name, model_type):

 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"}
 class FalModel():
     def __init__(self, model_name, model_type):