vinesmsuic commited on
Commit
530174e
·
1 Parent(s): 90372f3

Adding LTXVideo

Browse files
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"}