Skip to content

GitLab

Commit a81ceeb3 authored by mashun1's avatar mashun1
Browse files

Update animate_dist.py

parent 45d3f4ff
Hide 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,
...@@ -228,4 +228,4 @@ if __name__ == '__main__': ...@@ -228,4 +228,4 @@ if __name__ == '__main__':
parser.add_argument("--guidance_scale", type=float, default=7.5, required=False) parser.add_argument("--guidance_scale", type=float, default=7.5, required=False)
parser.add_argument("--save_path", type=str, default=None, required=True) parser.add_argument("--save_path", type=str, default=None, required=True)
args = parser.parse_args() args = parser.parse_args()
run(args) run(args)
\ 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