cbensimon HF staff commited on
Commit
5c1033f
1 Parent(s): 6d06601

Update demo/animate.py

Browse files
Files changed (1) hide show
  1. demo/animate.py +5 -3
demo/animate.py CHANGED
@@ -78,9 +78,11 @@ class MagicAnimate():
78
  controlnet.to(torch.float16)
79
  self.appearance_encoder.to(torch.float16)
80
 
81
- unet.enable_xformers_memory_efficient_attention()
82
- self.appearance_encoder.enable_xformers_memory_efficient_attention()
83
- controlnet.enable_xformers_memory_efficient_attention()
 
 
84
 
85
  self.pipeline = AnimationPipeline(
86
  vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet, controlnet=controlnet,
 
78
  controlnet.to(torch.float16)
79
  self.appearance_encoder.to(torch.float16)
80
 
81
+ ### Not yet compatible with ZeroGPU
82
+ # unet.enable_xformers_memory_efficient_attention()
83
+ # self.appearance_encoder.enable_xformers_memory_efficient_attention()
84
+ # controlnet.enable_xformers_memory_efficient_attention()
85
+ ###
86
 
87
  self.pipeline = AnimationPipeline(
88
  vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet, controlnet=controlnet,