6 Commits

Author	SHA1	Message	Date
David Ford	4fad71cd8c	Training optimizations (#217) ... * Optimizations to the training model Based on the changes made in textual_inversion I carried over the relevant changes that improve model training. These changes reduce the amount of memory used, significantly improve the speed at which training runs, and improves the quality of the results. It also fixes the problem where the model trainer wouldn't automatically stop when it hit the set number of steps. * Update main.py Cleaned up whitespace	2022-08-30 15:59:32 -04:00
Lincoln Stein	4f02b72c9c	prettified all the code using "blue" at the urging of @tildebyte	2022-08-26 03:15:42 -04:00
Sean McLellan	a93a79568d	Tweak save_top_k setting	2022-08-23 21:56:05 -04:00
Sean McLellan	8952196bbf	Add personalization	2022-08-23 18:26:28 -04:00
ablattmann	171cf29fb5	add configs for training unconditional/class-conditional ldms	2021-12-22 15:57:23 +01:00
ablattmann	e66308c7f2	add code	2021-12-21 03:23:41 +01:00