Spaces:
Configuration error
Configuration 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,
|