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