How to train the model?

[XiaominLi1997]

Hi, thanks for your nice work, if it is possible you will release the training code?

[ersanliqiao]

I have the same issue

[YunChangLi]

Is the training code same with the controlNet ?

[CharlesGong12]

+1

[761qgmpgz943]

Hi, thanks for your nice work, if it is possible you will release the training code?
It's okay not to organize the training code, I have some questions about the training process.

[vidit98]

Hi all, It will be difficult to release the training code. However, I am willing to help in specific doubts related to training. @YunChangLi yes the training code closely follows the controlnet repo. The major difference would be in dataloader where you also need to read the segmentation masks and in conditioning process. The conditioning code can be copied from the inference code already released.