update

README.md

@@ -7,5 +7,9 @@ sdk: gradio
 sdk_version: 4.4.0
 python_version: 3.8
 app_file: app.py
+models:
+ - zcxu-eric/MagicAnimate
+ - runwayml/stable-diffusion-v1-5
+ - stabilityai/sd-vae-ft-mse
 pinned: false
 ---

app.py

@@ -17,15 +17,6 @@ from subprocess import PIPE, run
 
 from demo.animate import MagicAnimate
 
-for command in [
-    'mkdir ./pretrained_models && cd pretrained_models',
-    'git lfs clone https://huggingface.co/zcxu-eric/MagicAnimate',
-    'git lfs clone https://huggingface.co/runwayml/stable-diffusion-v1-5',
-    'git lfs clone https://huggingface.co/stabilityai/sd-vae-ft-mse',
-    'cd ..',
-]:
-    run(command, stdout=PIPE, stderr=PIPE, universal_newlines=True, shell=True)
-
 animator = MagicAnimate()
 
 def animate(reference_image, motion_sequence_state, seed, steps, guidance_scale):

configs/prompts/animation.yaml

@@ -1,10 +1,10 @@
-pretrained_model_path: "pretrained_models/stable-diffusion-v1-5"
-pretrained_vae_path: "pretrained_models/sd-vae-ft-mse"
-pretrained_controlnet_path: "pretrained_models/MagicAnimate/densepose_controlnet"
-pretrained_appearance_encoder_path: "pretrained_models/MagicAnimate/appearance_encoder"
+pretrained_model_path: "stable-diffusion-v1-5"
+pretrained_vae_path: "sd-vae-ft-mse"
+pretrained_controlnet_path: "MagicAnimate/densepose_controlnet"
+pretrained_appearance_encoder_path: "MagicAnimate/appearance_encoder"
 pretrained_unet_path: ""
 
-motion_module: "pretrained_models/MagicAnimate/temporal_attention/temporal_attention.ckpt"
+motion_module: "MagicAnimate/temporal_attention/temporal_attention.ckpt"
 
 savename: null