JoePenna / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual Inversion (https://arxiv.org/abs/2208.01618) for Stable Diffusion (https://arxiv.org/abs/2112.10752). Tweaks focused on training faces, objects, and styles.
MIT License
3.19k stars 558 forks source link

"No training images provided" error #182

Closed jeekca closed 1 year ago

jeekca commented 1 year ago

Hey guys,

At the "Dreambooth Training Environment Setup" step, I click the orange "Training Images (0)" button, Windows' File Explorer pops up, I select all 26 of my PNG images, then once I press "Open" within the File Explorer, the section's form disappears. So in short, it goes from this:

dreambooth-img-upload-error_002

...to this:

dreambooth-img-upload-error_001

...after pressing "Open".

Here's some additional info in list-form:

I've updated the "Docker Image Name" to runpod/pytorch-3.10-1.13.1-116 as suggested here, but I still get the same exact bug.

Any insights?

Thanks a ton!

djbielejeski commented 1 year ago

Can you provide your 26 training images so we can reproduce? You can black them out or blur them or whatever, just keep the names the same.

These widgets are finicky, and other people have been reporting a few issues with them as well.

jeekca commented 1 year ago

Hey @djbielejeski,

Thanks for the prompt response.

I've attached the training images to this message. Hopefully this helps. I was still able to get my training done through the legacy dreambooth file, but hopefully I get to do my training on the new dreambooth file soon, as it's so much more user-friendly.

Thanks!

Link removed by author.

djbielejeski commented 1 year ago

Looks like the ipwidget FileUpload can only do up to 10mb max (which is incredibly stupid lol).

We have a couple work arounds

1) Upload 1 image and proceed as normal. After the config is generated all you have to do is upload your remaining training images to the training_images folder. 2) Compress your input images to stay under the 10mb file limit by using jpegs. You shouldn't notice any quality loss.

jeekca commented 1 year ago

@djbielejeski Thank you so much for the prompt assistance! Now that you mention it, my PNG-s are wildly uncompressed.

It'd be nice to have a little file size indicator or better error message, but at least I can work around the limitation now that I know the cause. I'll go with your suggestions.

Thanks a ton once again!

brunobpsrpg commented 1 year ago

Idk if it helps, but i always put my pngs on BIRME, this will compress and resize the files for you. Hope it helps.

https://www.birme.net/?target_width=512&target_height=512