lllyasviel / ControlNet

Let us control diffusion models!
Apache License 2.0
28.97k stars 2.62k forks source link

can unlocked diffuison model be saved? #608

Open BigDevil82 opened 6 months ago

BigDevil82 commented 6 months ago

Thanks for the great work!

I'm training ControlNet on my dataset, and setting sd_locked = False turns out to improve model performance significantly. I learned that unlocking sd will also train some parts of the diffusion model, so the base model also changed, right? But one of the problems is that I can not use it on automatical1111 sd webui, which I can use to combine other techniques like LoRA with ControlNet. Does it mean if I unlocked SD when training ControlNet, I have to use the official code base to take full use of the trained model, i.e. running in prediction mode? Is it possible to save the fine-tuned SD along with ControlNet so I can use them in other places together? Appreciate any advice.