Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
Commit
·
530174e
1
Parent(s):
90372f3
Adding LTXVideo
Browse files- model/model_registry.py +10 -0
- model/models/__init__.py +2 -1
model/model_registry.py
CHANGED
@@ -459,3 +459,13 @@ register_model_info(
|
|
459 |
"text2video_generation"
|
460 |
)
|
461 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
459 |
"text2video_generation"
|
460 |
)
|
461 |
|
462 |
+
register_model_info(
|
463 |
+
["videogenhub_LTXVideo_text2video"],
|
464 |
+
"LTXVideo",
|
465 |
+
"https://github.com/Lightricks/LTX-Video",
|
466 |
+
"DiT based Video Generation Model",
|
467 |
+
"Apache 2.0",
|
468 |
+
"Lightricks",
|
469 |
+
"text2video_generation"
|
470 |
+
)
|
471 |
+
|
model/models/__init__.py
CHANGED
@@ -29,7 +29,8 @@ VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
|
29 |
'fal_StableVideoDiffusion_text2video',
|
30 |
'fal_CogVideoX-5B_text2video',
|
31 |
'videogenhub_PyramidFlow_text2video',
|
32 |
-
'videogenhub_Allegro_text2video'
|
|
|
33 |
|
34 |
MAP_NAMES_IMAGENHUB = {}
|
35 |
MAP_NAMES_VIDEOGENHUB = {"CogVideoX-2B": "CogVideoX", "CogVideoX-5B": "CogVideoX5B"}
|
|
|
29 |
'fal_StableVideoDiffusion_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"}
|