Error while Training about dreambooth-stable-diffusion HOT 5 CLOSED

I went to line 896 on Main.py, and changed "trainer" for "Trainer", now it's working

from dreambooth-stable-diffusion.

Potentially related (NameError: name 'trainer' is not defined):

• #28
• #53
• #86
• #87

from dreambooth-stable-diffusion.

@H41907 were you able to get 1.5 to work ? im using the model from runwayml/stable-diffusion-v1-5 and it is constantly erroring out. could you point out how u got it to work ?

from dreambooth-stable-diffusion.

No, not resolved. I also stopped looking into it: at the time I was thinking that the inpainting model is different to the standard SD model (1.4/1.5) and therefore not useable for Dreambooth.
size mismatch for model.diffusion_model.input_blocks.0.0.weight: copying a param with shape torch.Size([320, 9, 3, 3]) from checkpoint, the shape in current model is torch.Size([320, 4, 3, 3]).

I did not continue looking into it, maybe nowadays there is a solution or tool which solves the issue.

from dreambooth-stable-diffusion.

@H41907 same here. i moved to the Diffuser way of doing things. got it to work well. actually the Huggingface "accelerate" libraries make training much faster with 1.5.

here's my simple notebook on this - https://github.com/sandys/SimpleDiffuserDreambooth

from dreambooth-stable-diffusion.