Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
Commit
•
fb2a667
1
Parent(s):
c9c2b46
Adding mochi
Browse files- model/model_registry.py +9 -0
- model/models/__init__.py +2 -1
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"}
|