Upload animate.py

demo/animate.py

@@ -66,13 +66,12 @@ class MagicAnimate():
         ### Load controlnet
         controlnet = ControlNetModel.from_pretrained(config.pretrained_controlnet_path)
 
-        vae.to(torch.float16)
-        unet.to(torch.float16)
-        text_encoder.to(torch.float16)
-        controlnet.to(torch.float16)
-        self.appearance_encoder.to(torch.float16)
-        
         if device == "cuda":
+            vae.to(torch.float16)
+            unet.to(torch.float16)
+            text_encoder.to(torch.float16)
+            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()