radames commited on
Commit
fe56d1b
1 Parent(s): bfc720c

make sure model paths match config.yaml

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -17,6 +17,12 @@ from subprocess import PIPE, run
17
 
18
  from demo.animate import MagicAnimate
19
 
 
 
 
 
 
 
20
  animator = MagicAnimate()
21
 
22
  def animate(reference_image, motion_sequence_state, seed, steps, guidance_scale):
 
17
 
18
  from demo.animate import MagicAnimate
19
 
20
+ from huggingface_hub import snapshot_download
21
+
22
+ snapshot_download(repo_id="runwayml/stable-diffusion-v1-5", local_dir="./stable-diffusion-v1-5")
23
+ snapshot_download(repo_id="stabilityai/sd-vae-ft-mse", local_dir="./sd-vae-ft-mse")
24
+ snapshot_download(repo_id="zcxu-eric/MagicAnimate", local_dir="./MagicAnimate")
25
+
26
  animator = MagicAnimate()
27
 
28
  def animate(reference_image, motion_sequence_state, seed, steps, guidance_scale):