From 5821523a524190490a287c5e2aacb6e72cc3a4cf Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 17 Jan 2023 07:20:45 +0100 Subject: Update --- train_dreambooth.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'train_dreambooth.py') diff --git a/train_dreambooth.py b/train_dreambooth.py index d722e68..48bdcf8 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py @@ -14,8 +14,7 @@ from slugify import slugify from util import load_config, load_embeddings_from_dir from data.csv import VlpnDataModule, keyword_filter -from training.functional import train, generate_class_images, add_placeholder_tokens, get_models -from training.strategy.ti import textual_inversion_strategy +from training.functional import train, get_models from training.strategy.dreambooth import dreambooth_strategy from training.optimization import get_scheduler from training.util import save_args @@ -610,7 +609,7 @@ def main(): ) trainer( - callbacks_fn=dreambooth_strategy, + strategy=dreambooth_strategy, project="dreambooth", train_dataloader=datamodule.train_dataloader, val_dataloader=datamodule.val_dataloader, -- cgit v1.2.3-54-g00ecf