Implementation Details

[frankkim1108]

Hello, @dunbar12138 and @linwansong

First of all, great work and thank you for making the code public.

I have some questions on how much computational power is needed to train and evaluate the model?

How much GPU VRAM is needed for training and evaluating the model?

Thank you for your time.

[dunbar12138]

Hi, thanks for your interest!

We train and evaluate our model on GPUs with 40GB VRAM, but our inference code should also be runnable on a 24GB GPU.

For training a ControlNet, you can refer to this link for more detailed specifications about GPU requirements. We recommend using a larger batch size to achieve a better convergence.