d8ahazard / sd_dreambooth_extension

Other
1.86k stars 282 forks source link

Skip images which failed to load #1418

Closed RossM closed 9 months ago

RossM commented 9 months ago

Describe your changes

An exception while reading an image (such as PIL.Image.DecompressionBombError) during precache causes the entire training run to fail with an unhelpful error that doesn't even say what file didn't load. Instead, skip images that didn't load.

Issue ticket number and link (if applicable)

Checklist before requesting a review