Commit Graph

11 Commits (main)

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
d8ahazard 7dfd69a5f9 Work sync 2023-05-23 08:27:11 -05:00
d8ahazard 58e5de19e1 New UI Fun 2023-05-16 13:00:48 -05:00
d8ahazard 071f7fc1a9 Dreambooth logging 2023-05-11 11:46:49 -05:00
d8ahazard 4069f01ad3 Defaults! 2023-05-10 15:20:46 -05:00
d8ahazard 6ea5b5e6a0 UI Updates 2023-05-09 10:10:55 -05:00