search

pixeli99 / SVD_Xtend

Stable Video Diffusion Training Code and Extensions.
481 stars 45 forks source link

Out of Memory #27

Closed AbrahamYabo closed 6 months ago

AbrahamYabo commented 6 months ago

How much GPU memory is used for resolution of 512*320 and batch size of 1?

pixeli99 commented 6 months ago

This depends on which parameters you have activated for training, if you fully activate unet, it might require around 60GB of GPU memory, therefore if you're encountering out-of-memory issues, it is recommended to train only a small subset of parameters.

AbrahamYabo commented 6 months ago

Thank you for your reply.