Commit Graph

6 Commits (573d1c92bcb4cdb675eb47f6cf7e86a6ebbbbaf1)

Author SHA1 Message Date
Ross Morgan-Linial 573d1c92bc Add freeze_spectral_norm option
See https://arxiv.org/abs/2303.06296

This adds an option to reparametrize the model weights using the spectral norm so that the overall norm of each weight can't change. This helps to stabilize training at high learning rates.
2024-01-17 09:24:37 -08:00
Ross Morgan-Linial 93a6b5dc5e Add slider to control DREAM lambda
This allows adjusting the tradeoff between faster learning and better composition vs better preservation of detail.
2023-12-23 14:37:39 -08:00
Ross Morgan-Linial 44a520be28 Implement DREAM
DREAM adds an additional forward pass during training to make the model more robust to errors in early sampling steps. See http://arxiv.org/abs/2312.00210
2023-12-22 15:12:28 -08:00
saunderez 225515a532 Lora Save Frequency Fix
8bit optiniser load fix (requires bitsandbyes 0.41.1)
2023-09-07 06:04:06 +10:00
d8ahazard 4acfe9ae48 Merge @saunderez optimizers/SDXL branch
For testing, ya know?
2023-09-05 20:29:12 -05:00
d8ahazard 8bba174139 New UI 2023-05-25 18:33:49 -05:00