search

jinxixiang / magic_animate_unofficial

An unoffical training code of Magic Animate
Apache License 2.0
35 stars 5 forks source link

OOM during infer #12

Closed WilliamZhaoz closed 8 months ago

WilliamZhaoz commented 8 months ago

Thanks for your great efforts for optimize the magic animate and share the training code. When I run the infer script in the repo, I hit a OOM error (I have 80G memory), but when I run infer script in official repo, it works well (only take about 15G memory), do you have any idea about the problem? Looking forward to your reply, thanks!

jinxixiang commented 8 months ago

Thank you. Please make sure you have torch==2.0.0+xformers==0.0.19.

I used v100 for inference and it takes 14G GPU memory.

WilliamZhaoz commented 8 months ago

Thank you. Please make sure you have torch==2.0.0+xformers==0.0.19.

I used v100 for inference and it takes 14G GPU memory.

Thanks for your quick reply! It works after I update my torch version.