InvokeAI/configs
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
..
autoencoder add code 2021-12-21 03:23:41 +01:00
latent-diffusion stable diffusion 2022-08-10 16:30:49 +02:00
retrieval-augmented-diffusion stable diffusion 2022-08-10 16:30:49 +02:00
stable-diffusion Training optimizations (#217) 2022-08-30 15:59:32 -04:00