How to train ControlNet-XS with my own dataset?

[hackmebroo]

Great Job! How to train ControlNet-XS with my own dataset?

[Whoo-jl]

I have the same problem

[universewill]

same problem here

[zdxpan]

i`m trying , only succ on sd2.1
but it is not yet refined enough.
We are still in the process of experimenting with training on SDXL and making modifications to the input, data shape, format, and other areas.

We have initiated a pull request (PR) and would appreciate everyone's feedback and collaboration to further improve the model. Let's engage in more discussions and exchanges.

#13

[universewill]

any help here？ @Sipirius

[Sipirius]

The initial project was purely for inference and testing. However, we will probably release the training scripts at some point next month.

Cheers

[shoutOutYangJie]

The initial project was purely for inference and testing. However, we will probably release the training scripts at some point next month.

Cheers

do you plan open the training code?

[ifsheldon]

+1. Huge boost for the community if you can open-source the training script.

[aseemkhanduja]

Will be great to have the training scripts please :)

[oxysoft]

+1 also wondering

[Sipirius]

The training script will be released approximately next week.

cheers

[Sipirius]

Training functionality is added to the repo - Readme Addition

You would need to adapt the example dataset ldm.data.dummy_set.DummyBase to your specific needs.

cheers