Spaces:
Running
on
Zero
Running
on
Zero
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"}
|