Spaces:
Running
on
Zero
Running
on
Zero
Merge branch 'main' of https://huggingface.co/spaces/TIGER-Lab/GenAI-Arena
Browse files- model/models/__init__.py +8 -4
model/models/__init__.py
CHANGED
|
@@ -4,14 +4,18 @@ from .fal_api_models import load_fal_model
|
|
| 4 |
from .videogenhub_models import load_videogenhub_model
|
| 5 |
|
| 6 |
|
| 7 |
-
IMAGE_GENERATION_MODELS = ['fal_LCM(v1.5/XL)_text2image','fal_SDXLTurbo_text2image','fal_SDXL_text2image', 'imagenhub_PixArtAlpha_generation', 'fal_PixArtSigma_text2image',
|
| 8 |
-
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
| 10 |
IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
|
| 11 |
'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
|
| 12 |
'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',
|
| 13 |
'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition']
|
| 14 |
-
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
|
|
|
| 15 |
#'fal_StableVideoDiffusion_text2video',
|
| 16 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|
| 17 |
'videogenhub_ModelScope_generation', 'videogenhub_OpenSora_generation']
|
|
|
|
| 4 |
from .videogenhub_models import load_videogenhub_model
|
| 5 |
|
| 6 |
|
| 7 |
+
# IMAGE_GENERATION_MODELS = ['fal_LCM(v1.5/XL)_text2image','fal_SDXLTurbo_text2image','fal_SDXL_text2image', 'imagenhub_PixArtAlpha_generation', 'fal_PixArtSigma_text2image',
|
| 8 |
+
# 'imagenhub_OpenJourney_generation','fal_SDXLLightning_text2image', 'fal_StableCascade_text2image',
|
| 9 |
+
# 'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
|
| 10 |
+
IMAGE_GENERATION_MODELS = ['imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation', 'imagenhub_PixArtAlpha_generation',
|
| 11 |
+
'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation',
|
| 12 |
+
'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
|
| 13 |
IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
|
| 14 |
'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
|
| 15 |
'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',
|
| 16 |
'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition']
|
| 17 |
+
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
| 18 |
+
'fal_AnimateDiffTurbo_text2video',
|
| 19 |
#'fal_StableVideoDiffusion_text2video',
|
| 20 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|
| 21 |
'videogenhub_ModelScope_generation', 'videogenhub_OpenSora_generation']
|