Update animate.py

sdbds · Dec 6, 2023 · f8fb4cb · f8fb4cb
1 parent 419a6f0
commit f8fb4cb
Showing 1 changed file with 1 addition and 1 deletion.
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 :
+        if config.pretrained_vae_path:
             vae = AutoencoderKL.from_pretrained(config.pretrained_vae_path)
         else:
             vae = AutoencoderKL.from_pretrained(config.pretrained_model_path, subfolder="vae")