test upload

model/model_registry.py

@@ -220,6 +220,13 @@ register_model_info(
 #    "https://arxiv.org/abs/2205.15868",
 #    "Text-to-Video Generation via Transformers",
 #)
+
+register_model_info(
+    ["videogenhub_OpenSora_generation"],
+    "OpenSora",
+    "videogenhub_OpenSora_generation",
+    "A community-driven opensource implementation of Sora.",
+)
     
 
 models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation',
@@ -228,4 +235,5 @@ models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub
           'imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
           'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',
           "fal_AnimateDiffTurbo_text2video", "fal_AnimateDiff_text2video",
-          "videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation"]
+          "videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation",
+          "videogenhub_OpenSora_generation"]

model/models/__init__.py

@@ -13,7 +13,7 @@ IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZe
 VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
                            'fal_AnimateDiffTurbo_text2video',
                            'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
-                           'videogenhub_ModelScope_generation']
+                           'videogenhub_ModelScope_generation', 'videogenhub_OpenSora_generation']
 
 
 def load_pipeline(model_name):

requirements.txt

@@ -59,4 +59,13 @@ protobuf==3.20.*
 rotary_embedding_torch
 av
 natsort
-
+#for opensora
+colossalai
+mmengine
+pre-commit
+pyarrow
+pyav
+tensorboard
+timm
+wandb
+pandarallel