diff --git a/configs/prompts/animation.yaml b/configs/prompts/animation.yaml index d4c15903..6aab720d 100644 --- 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 index b71f1940..696b7b15 100644 --- 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")