d8ahazard / sd_dreambooth_extension

Other
1.85k stars 283 forks source link

Rearrange UI, Add quite a bit of features, bring back min SNR #1466

Closed Anzhc closed 2 months ago

Anzhc commented 4 months ago

Describe your changes

  1. Re-arranged UI to make more sense from trainer pov.
  2. Added bunch of features from my older fork that were tested thoroughly by few trainers on it, and provided benefit, in one way or another. Those features are: min snr (idk why it disappeared in rework, @saunderez added it while back) debiased estimation loss multires noise ip noise gamma loss curve offset (my own thingie) Scheduled noise offset (Shown to be especially beneficial in longer trainings) Randomized noise offset

Issue ticket number and link (if applicable)

Not applicable

Checklist before requesting a review

Here is example, just so you can be sure i actually did train xD All are underbaked, because i used twice less steps than needed. Loras don't work in post-XL rework after all, gotta do dreambooth... изображение First is all features on, second is pre-noise-stuff added and fixed min snr, third is zero min snr

P.S.S. Yes, i have no idea how to use version control and updating whole files. Here is some screenshots of how it looks right now: First, order of tabs(I touched only Parameters UI): изображение Let's look at accordions now. Intervals were not changed. Lora was minified a bit with row. Batch size and image size were merged in single tab, alongside with their utilities. изображение Learning rate tab was only moved. Performance was strongly reworked, and extended with new features. изображение Prior loss was minified too. изображение Saving, Image generation and Extras were not touched. Now, whole basic UI is really small, but i probably f'ed up here and made it hide some relevant stuff? Idk, i don't recall how it looked before that tbh. изображение