tianleliphoebe commited on
Commit
9d5a8fc
1 Parent(s): 4778760
Files changed (1) hide show
  1. 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
- 'imagenhub_OpenJourney_generation','fal_SDXLLightning_text2image', 'fal_StableCascade_text2image',
9
- 'playground_PlayGroundV2_generation', 'playground_PlayGroundV2.5_generation']
 
 
 
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', 'fal_AnimateDiffTurbo_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']