remove vae path

sdbds · Dec 5, 2023 · 1b8911a · 1b8911a
1 parent 180e6c5
commit 1b8911a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/prompts/animation.yaml b/configs/prompts/animation.yaml
@@ -1,5 +1,5 @@
 pretrained_model_path: "pretrained_models/stable-diffusion-v1-5"
-pretrained_vae_path: "pretrained_models/sd-vae-ft-mse"
+pretrained_vae_path: ""
 pretrained_controlnet_path: "pretrained_models/MagicAnimate/densepose_controlnet"
 pretrained_appearance_encoder_path: "pretrained_models/MagicAnimate/appearance_encoder"
 pretrained_unet_path: ""

diff --git a/demo/animate.py b/demo/animate.py
@@ -64,7 +64,7 @@ def __init__(self, config="configs/prompts/animation.yaml") -> None:
         self.appearance_encoder = AppearanceEncoderModel.from_pretrained(config.pretrained_appearance_encoder_path, subfolder="appearance_encoder").cuda()
         self.reference_control_writer = ReferenceAttentionControl(self.appearance_encoder, do_classifier_free_guidance=True, mode='write', fusion_blocks=config.fusion_blocks)
         self.reference_control_reader = ReferenceAttentionControl(unet, do_classifier_free_guidance=True, mode='read', fusion_blocks=config.fusion_blocks)
-        if config.pretrained_vae_path is not None:
+        if config.pretrained_vae_path :
             vae = AutoencoderKL.from_pretrained(config.pretrained_vae_path)
         else:
             vae = AutoencoderKL.from_pretrained(config.pretrained_model_path, subfolder="vae")