d8ahazard / sd_dreambooth_extension

Other
1.85k stars 281 forks source link

Skip broken image files #1416

Closed RossM closed 7 months ago

RossM commented 7 months ago

Describe your changes

Currently, if there is an error while reading an image during precache, such as due to PIL.Image.DecompressionBombError, the entire training run fails with an unhelpful error that doesn't even say which file failed. This patch changes the loader to skip images which can't be loaded.

Issue ticket number and link (if applicable)

Checklist before requesting a review