fix training scripts except controlnet not working

This commit is contained in:
Kohya S
2023-06-22 08:46:53 +09:00
parent 1c09867b3e
commit 5114e8daf1
7 changed files with 14 additions and 9 deletions

View File

@@ -151,7 +151,7 @@ def train(args):
# データセットを準備する
if args.dataset_class is None:
blueprint_generator = BlueprintGenerator(ConfigSanitizer(True, True, False))
blueprint_generator = BlueprintGenerator(ConfigSanitizer(True, True, False, False))
if args.dataset_config is not None:
accelerator.print(f"Load dataset config from {args.dataset_config}")
user_config = config_util.load_user_config(args.dataset_config)