GPU Memory of training model

[YangXiuyu]

Hi, could you provide some information about your gpu memory consumption? I notice that you used two controlnets, two unets and a nerf model at the same time, while I feel that requires a lot of memory. Thanks

[Fan Lu]

Hi, thanks for your interests.
We train the model using a single NVIDIA V100 GPU (32GB).