yuanshengni commited on
Commit
765fb5e
1 Parent(s): b1657de

update fal_svd

Browse files
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(