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
Sean McLellan	7081a84600	Fix GPU parameter in readme, add SD style finetune config	2022-08-23 20:14:40 -04:00
Sean McLellan	8952196bbf	Add personalization	2022-08-23 18:26:28 -04:00
Robin Rombach	2ff270f4e0	stable diffusion	2022-08-10 16:30:49 +02: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