ProFusion icon indicating copy to clipboard operation
ProFusion copied to clipboard

Training on SD 1.5

Open shenzy08 opened this issue 2 years ago • 1 comments

Hello, I would like to know that whether I can train this model on stable-diffusion-v1-5, which is applicable to ControlNet. I have tried to change the base model, but encounter the dimension mismatch error, and it cannot be addressed simply like changing the clip model. Please offer me some help, thank you!

shenzy08 avatar May 26 '23 10:05 shenzy08

Yes, you need to change the clip model, and also change some dimension arguments. For example, because SD v1.5 and SD v2 are based on different text encoders, you need to change the output dimension. You may also want to change the dimension of projection if you change CLIP image encoder.

drboog avatar May 26 '23 15:05 drboog