Upload animate.py

demo/animate.py

@@ -72,9 +72,10 @@ 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()
+        if device == "cuda":
+            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,