vinesmsuic commited on
Commit
222ecc0
1 Parent(s): 623aaf3

add CosXL and InfEdit

Browse files
Files changed (1) hide show
  1. 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', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_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',