AnimateDiff icon indicating copy to clipboard operation
AnimateDiff copied to clipboard

Training Cost

Open AmberCheng opened this issue 1 year ago • 1 comments

Hi, I am currently also fine-tuning the motion module of Animatediff. However, I am using the original GitHub(https://github.com/guoyww/AnimateDiff). I found that when the training size is 512x512x16 frames, the GPU memory has reached 51G. I wonder what are your training configurations and resource consumption? Looking forward to your reply~

AmberCheng avatar Aug 23 '24 09:08 AmberCheng

Thats too many, 512x512x16 must took somewhere 15-16gb, I guess you didn't turn off gradients on unet. You must turn on grandients only on motion module.

tumurzakov avatar Aug 26 '24 07:08 tumurzakov