vinesmsuic commited on
Commit
4e37c8d
1 Parent(s): b3e524c

adding CogVideoX and OpenSora

Browse files
model/model_registry.py CHANGED
@@ -285,6 +285,20 @@ register_model_info(
285
  "https://github.com/hpcaitech/Open-Sora",
286
  "A community-driven opensource implementation of Sora.",
287
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
288
 
289
  register_model_info(
290
  ["videogenhub_T2VTurbo_generation"],
 
285
  "https://github.com/hpcaitech/Open-Sora",
286
  "A community-driven opensource implementation of Sora.",
287
  )
288
+
289
+ register_model_info(
290
+ ["videogenhub_OpenSora12_generation"],
291
+ "OpenSora v1.2",
292
+ "https://github.com/hpcaitech/Open-Sora",
293
+ "A community-driven opensource implementation of Sora. v1.2",
294
+ )
295
+
296
+ register_model_info(
297
+ ["videogenhub_CogVideoX_generation"],
298
+ "CogVideoX",
299
+ "https://github.com/THUDM/CogVideo",
300
+ "Text-to-Video Diffusion Models with An Expert Transformer.",
301
+ )
302
 
303
  register_model_info(
304
  ["videogenhub_T2VTurbo_generation"],
model/models/__init__.py CHANGED
@@ -19,7 +19,7 @@ VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
19
  'fal_AnimateDiffTurbo_text2video',
20
  'videogenhub_LaVie_generation',
21
  'videogenhub_VideoCrafter2_generation',
22
- 'videogenhub_ModelScope_generation',
23
  'videogenhub_OpenSora_generation', 'videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
24
  MUSEUM_UNSUPPORTED_MODELS = ['videogenhub_OpenSoraPlan_generation']
25
  DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
 
19
  'fal_AnimateDiffTurbo_text2video',
20
  'videogenhub_LaVie_generation',
21
  'videogenhub_VideoCrafter2_generation',
22
+ 'videogenhub_ModelScope_generation', 'videogenhub_CogVideoX_generation', 'videogenhub_OpenSora12_generation'
23
  'videogenhub_OpenSora_generation', 'videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
24
  MUSEUM_UNSUPPORTED_MODELS = ['videogenhub_OpenSoraPlan_generation']
25
  DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']