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

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