vinesmsuic commited on
Commit
fb2a667
1 Parent(s): c9c2b46

Adding mochi

Browse files
model/model_registry.py CHANGED
@@ -469,3 +469,12 @@ register_model_info(
469
  "text2video_generation"
470
  )
471
 
 
 
 
 
 
 
 
 
 
 
469
  "text2video_generation"
470
  )
471
 
472
+ register_model_info(
473
+ ["videogenhub_Mochi1_text2video"],
474
+ "Mochi1",
475
+ "https://github.com/genmoai/mochi",
476
+ "Mochi 1 preview is an open state-of-the-art video generation model with high-fidelity motion and strong prompt adherence in preliminary evaluation.",
477
+ "Apache 2.0",
478
+ "Genmo AI",
479
+ "text2video_generation"
480
+ )
model/models/__init__.py CHANGED
@@ -30,7 +30,8 @@ VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
30
  'fal_CogVideoX-5B_text2video',
31
  'videogenhub_PyramidFlow_text2video',
32
  'videogenhub_Allegro_text2video',
33
- 'videogenhub_LTXVideo_text2video']
 
34
 
35
  MAP_NAMES_IMAGENHUB = {}
36
  MAP_NAMES_VIDEOGENHUB = {"CogVideoX-2B": "CogVideoX", "CogVideoX-5B": "CogVideoX5B"}
 
30
  'fal_CogVideoX-5B_text2video',
31
  'videogenhub_PyramidFlow_text2video',
32
  'videogenhub_Allegro_text2video',
33
+ 'videogenhub_LTXVideo_text2video',
34
+ 'videogenhub_Mochi1_text2video']
35
 
36
  MAP_NAMES_IMAGENHUB = {}
37
  MAP_NAMES_VIDEOGENHUB = {"CogVideoX-2B": "CogVideoX", "CogVideoX-5B": "CogVideoX5B"}