TheLastBen / fast-stable-diffusion

fast-stable-diffusion + DreamBooth
MIT License
7.49k stars 1.31k forks source link

fine tuning on SD2-512. Got all brown images #629

Open jgeibel opened 1 year ago

jgeibel commented 1 year ago

Just saw that SD2-512 was an option, so I did my first training with a dataset I've used on 1.5 many times. When inferencing from that, all I get are brown images (with some black shadows). Curious if I did something wrong?

TheLastBen commented 1 year ago

the problem is with the inference, not the training, the 768 model works fine, but the 512 gives brown images, I'm trying to fin the cause

jgeibel commented 1 year ago

👍

archimedesinstitute commented 1 year ago

I just actually tried this with 768 and i got the brown. has 512 been fixed?

nmeln commented 1 year ago

Could this be issue with how cpkt is converted or read in?

When manually importing the model weights (fine-tuned on sd2.0-base), using StableDiffusionPipeline & different schedulers from diffusers package, it generates just fine

TheLastBen commented 1 year ago

Could this be issue with how cpkt is converted or read in?

very likely, yes, I don't think diffusers has the right conversion scripts yet.

archimedesinstitute commented 1 year ago

Can someone confirm I can do that in colab? I’m still getting brown outputs and I am really unclear on how to run it through correctly. Thanks!

Tophness commented 1 year ago

I got brown outputs with 512 and 768 ckpts, tried running the 768 diffusers output with the official sd 2.0 diffusers pipeline and still the same

guumaster commented 1 year ago

Any update on this? I have nice images after training, but loading ckpt gets all brown

TheLastBen commented 1 year ago

@guumaster that problem was solved days ago, use the latest colab and make sure you choose the correct model version in the cells

guumaster commented 1 year ago

I'm using the colab from today (3-4 hours ago), I've chosen the V2-768px version on both dropdown menus

Training and testing is ok. But if I take ckpt to another colab, I get the brown images. Any clues?

TheLastBen commented 1 year ago

Which other colab, fast-A1111 ? or not from this repo ?