Update animate_dist.py

demo/animate_dist.py

@@ -77,9 +77,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,
@@ -228,4 +228,4 @@ if __name__ == '__main__':
     parser.add_argument("--guidance_scale", type=float, default=7.5, required=False)
     parser.add_argument("--save_path", type=str, default=None, required=True)
     args = parser.parse_args()
-    run(args)
\ No newline at end of file
+    run(args)