vinesmsuic commited on
Commit
f848c71
1 Parent(s): e39ed79

remove cog video

Browse files
model/model_registry.py CHANGED
@@ -214,12 +214,12 @@ register_model_info(
214
  "ModelScope is a a T2V synthesis model that evolves from a T2I synthesis model.",
215
  )
216
 
217
- register_model_info(
218
- ["videogenhub_CogVideo_generation"],
219
- "CogVideo",
220
- "https://arxiv.org/abs/2205.15868",
221
- "Text-to-Video Generation via Transformers",
222
- )
223
 
224
 
225
  models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation',
@@ -228,4 +228,4 @@ models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub
228
  'imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
229
  'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',
230
  "fal_AnimateDiffTurbo_text2video", "fal_AnimateDiff_text2video",
231
- "videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation", "videogenhub_CogVideo_generation"]
 
214
  "ModelScope is a a T2V synthesis model that evolves from a T2I synthesis model.",
215
  )
216
 
217
+ #register_model_info(
218
+ # ["videogenhub_CogVideo_generation"],
219
+ # "CogVideo",
220
+ # "https://arxiv.org/abs/2205.15868",
221
+ # "Text-to-Video Generation via Transformers",
222
+ #)
223
 
224
 
225
  models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation',
 
228
  'imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
229
  'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',
230
  "fal_AnimateDiffTurbo_text2video", "fal_AnimateDiff_text2video",
231
+ "videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation"]
model/models/__init__.py CHANGED
@@ -13,7 +13,7 @@ IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZe
13
  VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
14
  'fal_AnimateDiffTurbo_text2video',
15
  'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
16
- 'videogenhub_ModelScope_generation', 'videogenhub_CogVideo_generation']
17
 
18
 
19
  def load_pipeline(model_name):
 
13
  VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
14
  'fal_AnimateDiffTurbo_text2video',
15
  'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
16
+ 'videogenhub_ModelScope_generation']
17
 
18
 
19
  def load_pipeline(model_name):