Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
Commit
•
f848c71
1
Parent(s):
e39ed79
remove cog video
Browse files- model/model_registry.py +7 -7
- model/models/__init__.py +1 -1
model/model_registry.py
CHANGED
@@ -214,12 +214,12 @@ register_model_info(
|
|
214 |
"ModelScope is a a T2V synthesis model that evolves from a T2I synthesis model.",
|
215 |
)
|
216 |
|
217 |
-
register_model_info(
|
218 |
-
["videogenhub_CogVideo_generation"],
|
219 |
-
"CogVideo",
|
220 |
-
"https://arxiv.org/abs/2205.15868",
|
221 |
-
"Text-to-Video Generation via Transformers",
|
222 |
-
)
|
223 |
|
224 |
|
225 |
models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation',
|
@@ -228,4 +228,4 @@ models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub
|
|
228 |
'imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
|
229 |
'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',
|
230 |
"fal_AnimateDiffTurbo_text2video", "fal_AnimateDiff_text2video",
|
231 |
-
"videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation"
|
|
|
214 |
"ModelScope is a a T2V synthesis model that evolves from a T2I synthesis model.",
|
215 |
)
|
216 |
|
217 |
+
#register_model_info(
|
218 |
+
# ["videogenhub_CogVideo_generation"],
|
219 |
+
# "CogVideo",
|
220 |
+
# "https://arxiv.org/abs/2205.15868",
|
221 |
+
# "Text-to-Video Generation via Transformers",
|
222 |
+
#)
|
223 |
|
224 |
|
225 |
models = ['imagenhub_LCM_generation','imagenhub_SDXLTurbo_generation','imagenhub_SDXL_generation',
|
|
|
228 |
'imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZero_edition', 'imagenhub_Prompt2prompt_edition',
|
229 |
'imagenhub_SDEdit_edition', 'imagenhub_InstructPix2Pix_edition', 'imagenhub_MagicBrush_edition', 'imagenhub_PNP_edition', 'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition',
|
230 |
"fal_AnimateDiffTurbo_text2video", "fal_AnimateDiff_text2video",
|
231 |
+
"videogenhub_LaVie_generation", "videogenhub_VideoCrafter2_generation", "videogenhub_ModelScope_generation"]
|
model/models/__init__.py
CHANGED
@@ -13,7 +13,7 @@ IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZe
|
|
13 |
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
14 |
'fal_AnimateDiffTurbo_text2video',
|
15 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|
16 |
-
'videogenhub_ModelScope_generation'
|
17 |
|
18 |
|
19 |
def load_pipeline(model_name):
|
|
|
13 |
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
14 |
'fal_AnimateDiffTurbo_text2video',
|
15 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|
16 |
+
'videogenhub_ModelScope_generation']
|
17 |
|
18 |
|
19 |
def load_pipeline(model_name):
|