From 3e7fbb7dce321435bbbb81361debfbc499bf9231 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 13 Jan 2023 22:25:30 +0100 Subject: Reverted modularization mostly --- train_dreambooth.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'train_dreambooth.py') 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 from util import load_config, load_embeddings_from_dir from pipelines.stable_diffusion.vlpn_stable_diffusion import VlpnStableDiffusion from data.csv import VlpnDataModule, VlpnDataItem -from training.common import loss_step, train_loop, generate_class_images, get_scheduler +from training.common import loss_step, train_loop, generate_class_images +from training.optimization import get_scheduler from training.lr import LRFinder from training.util import CheckpointerBase, save_args from models.clip.embeddings import patch_managed_embeddings -- cgit v1.2.3-54-g00ecf