stable-dreambooth icon indicating copy to clipboard operation
stable-dreambooth copied to clipboard

inference script just reconstructs training images

Open jhurlbut opened this issue 2 years ago • 1 comments

I was able to successfully train a model after chaning the diffusers version and changing batch size to 2 but when running inference on the output I only get reconstructions of the training images

jhurlbut avatar Sep 23 '22 07:09 jhurlbut

The reason may be the training iterations are too much, you can decrease the iterations by reducing the number of sampled images or early stopping. Generally about ~200 iterations are ok.

Victarry avatar Sep 23 '22 08:09 Victarry