Save weights/Generate samples improvements - Githubissues

d8ahazard / sd_dreambooth_extension

Other

1.85k stars 283 forks source link

Save weights/Generate samples improvements #1443

Closed RossM closed 5 months ago

RossM commented 5 months ago

Describe your changes

"Save weights" and "Generate samples" happen immediately
Generating samples won't try to load every image in the dataset when "[filewords]" is used
Generating samples handles exceptions reading images for "[filewords]"
Disables safety checker when generating samples

Issue ticket number and link (if applicable)

Checklist before requesting a review

[ ] This is based on the /dev branch (Or a fork of it)
[+] This was created or at least validated using a proper IDE
[+] I have tested this code and validated any modified functions
[N/A] I have added the appropriate documentation and hint strings if adding or changing a user-facing feature