Spaces:
Running
on
Zero
Running
on
Zero
yuanshengni
commited on
Commit
•
765fb5e
1
Parent(s):
b1657de
update fal_svd
Browse files- model/model_registry.py +7 -0
- model/models/__init__.py +1 -0
- model/models/fal_api_models.py +2 -0
model/model_registry.py
CHANGED
@@ -194,6 +194,13 @@ register_model_info(
|
|
194 |
"AnimateDiff is a text-driven models that produce diverse and personalized animated images.",
|
195 |
)
|
196 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
197 |
register_model_info(
|
198 |
["fal_AnimateDiffTurbo_text2video"],
|
199 |
"AnimateDiff Turbo",
|
|
|
194 |
"AnimateDiff is a text-driven models that produce diverse and personalized animated images.",
|
195 |
)
|
196 |
|
197 |
+
register_model_info(
|
198 |
+
["fal_StableVideoDiffusion_text2video"],
|
199 |
+
"StableVideoDiffusion",
|
200 |
+
"https://fal.ai/models/fal-ai/fast-svd/text-to-video/api",
|
201 |
+
"Stable Video Diffusion empowers individuals to transform text and image inputs into vivid scenes.",
|
202 |
+
)
|
203 |
+
|
204 |
register_model_info(
|
205 |
["fal_AnimateDiffTurbo_text2video"],
|
206 |
"AnimateDiff Turbo",
|
model/models/__init__.py
CHANGED
@@ -13,6 +13,7 @@ IMAGE_EDITION_MODELS = ['imagenhub_CycleDiffusion_edition', 'imagenhub_Pix2PixZe
|
|
13 |
'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition']
|
14 |
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
15 |
'fal_AnimateDiffTurbo_text2video',
|
|
|
16 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|
17 |
'videogenhub_ModelScope_generation', 'videogenhub_OpenSora_generation']
|
18 |
|
|
|
13 |
'imagenhub_InfEdit_edition', 'imagenhub_CosXLEdit_edition']
|
14 |
VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
15 |
'fal_AnimateDiffTurbo_text2video',
|
16 |
+
'fal_StableVideoDiffusion_text2video',
|
17 |
'videogenhub_LaVie_generation', 'videogenhub_VideoCrafter2_generation',
|
18 |
'videogenhub_ModelScope_generation', 'videogenhub_OpenSora_generation']
|
19 |
|
model/models/fal_api_models.py
CHANGED
@@ -77,6 +77,8 @@ class FalModel():
|
|
77 |
fal_model_name = 'fast-animatediff/text-to-video'
|
78 |
elif self.model_name == 'AnimateDiffTurbo':
|
79 |
fal_model_name = 'fast-animatediff/turbo/text-to-video'
|
|
|
|
|
80 |
else:
|
81 |
raise NotImplementedError(f"text2video model of {self.model_name} in fal is not implemented yet")
|
82 |
handler = fal_client.submit(
|
|
|
77 |
fal_model_name = 'fast-animatediff/text-to-video'
|
78 |
elif self.model_name == 'AnimateDiffTurbo':
|
79 |
fal_model_name = 'fast-animatediff/turbo/text-to-video'
|
80 |
+
elif self.model_name == 'StableVideoDiffusion':
|
81 |
+
fal_model_name = 'fal-ai/fast-svd/text-to-video'
|
82 |
else:
|
83 |
raise NotImplementedError(f"text2video model of {self.model_name} in fal is not implemented yet")
|
84 |
handler = fal_client.submit(
|