diff options
Diffstat (limited to 'train_dreambooth.py')
-rw-r--r-- | train_dreambooth.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index c892ebf..2145e2b 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
@@ -21,7 +21,8 @@ from slugify import slugify | |||
21 | from util import load_config, load_embeddings_from_dir | 21 | from util import load_config, load_embeddings_from_dir |
22 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion | 22 | from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion |
23 | from data.csv import VlpnDataModule, VlpnDataItem | 23 | from data.csv import VlpnDataModule, VlpnDataItem |
24 | from training.common import loss_step, train_loop, generate_class_images, get_scheduler | 24 | from training.common import loss_step, train_loop, generate_class_images |
25 | from training.optimization import get_scheduler | ||
25 | from training.lr import LRFinder | 26 | from training.lr import LRFinder |
26 | from training.util import CheckpointerBase, save_args | 27 | from training.util import CheckpointerBase, save_args |
27 | from models.clip.embeddings import patch_managed_embeddings | 28 | from models.clip.embeddings import patch_managed_embeddings |