Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
Commit
•
222ecc0
1
Parent(s):
623aaf3
add CosXL and InfEdit
Browse files- model/models/__init__.py +3 -1
model/models/__init__.py
CHANGED
@@ -7,7 +7,9 @@ IMAGE_GENERATION_MODELS = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_gener
|
|
7 |
'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation',
|
8 |
'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
|
9 |
IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
|
10 |
-
'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
|
|
|
|
|
11 |
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
12 |
'fal_AnimateDiffTurbo_text2video',
|
13 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|
|
|
7 |
'imagenhub_OpenJourney_generation','imagenhub_SDXLLightning_generation', 'imagenhub_StableCascade_generation',
|
8 |
'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
|
9 |
IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
|
10 |
+
'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition',
|
11 |
+
'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition',
|
12 |
+
'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition']
|
13 |
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
14 |
'fal_AnimateDiffTurbo_text2video',
|
15 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|