Update animate.py

45d3f4ff · mashun1 · e81904b8 · 45d3f4ff
Commit 45d3f4ff authored Dec 18, 2023 by mashun1
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

demo/animate.py demo/animate.py +4 -4

No files found.
--- a/demo/animate.py
+++ b/demo/animate.py
@@ -78,9 +78,9 @@ class MagicAnimate():
        controlnet.to(torch.float16)
        self.appearance_encoder.to(torch.float16)
        
-        unet.enable_xformers_memory_efficient_attention()
-        self.appearance_encoder.enable_xformers_memory_efficient_attention()
-        controlnet.enable_xformers_memory_efficient_attention()
+        # unet.enable_xformers_memory_efficient_attention()
+        # self.appearance_encoder.enable_xformers_memory_efficient_attention()
+        # controlnet.enable_xformers_memory_efficient_attention()

        self.pipeline = AnimationPipeline(
            vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet, controlnet=controlnet,