Skip to content

GitLab

Commit 45d3f4ff authored by mashun1's avatar mashun1
Browse files

Update animate.py

parent e81904b8
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
demo/animate.py
View file @ 45d3f4ff
...@@ -78,9 +78,9 @@ class MagicAnimate(): ...@@ -78,9 +78,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,
...@@ -192,4 +192,4 @@ class MagicAnimate(): ...@@ -192,4 +192,4 @@ class MagicAnimate():
save_videos_grid(samples_per_video, animation_path) save_videos_grid(samples_per_video, animation_path)
return animation_path return animation_path
\ No newline at end of file
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