Spaces:
Runtime error
Runtime error
Update demo/animate.py
Browse files- 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 |
-
|
82 |
-
|
83 |
-
|
|
|
|
|
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,
|