Skip to content

GitLab

Commit a81ceeb3 authored by mashun1's avatar mashun1
Browse files

Update animate_dist.py

parent 45d3f4ff
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
demo/animate_dist.py
View file @ a81ceeb3
...@@ -77,9 +77,9 @@ class MagicAnimate(): ...@@ -77,9 +77,9 @@ class MagicAnimate():
controlnet.to(torch.float16) controlnet.to(torch.float16)
self.appearance_encoder.to(torch.float16) self.appearance_encoder.to(torch.float16)
unet.enable_xformers_memory_efficient_attention() # unet.enable_xformers_memory_efficient_attention()
self.appearance_encoder.enable_xformers_memory_efficient_attention() # self.appearance_encoder.enable_xformers_memory_efficient_attention()
controlnet.enable_xformers_memory_efficient_attention() # controlnet.enable_xformers_memory_efficient_attention()
self.pipeline = AnimationPipeline( self.pipeline = AnimationPipeline(
vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet, controlnet=controlnet, vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet, controlnet=controlnet,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment