VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-17 17:24:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

change dataset config option to dataset_config

Browse Source
This commit is contained in:
Kohya S
2023-03-02 21:51:58 +09:00
parent 2d2407410e
commit 5602e0e5fc
6 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
train_network.py
View File

@@ -54,7 +54,7 @@ def train(args):
cache_latents = args.cache_latents
use_dreambooth_method = args.in_json is None
use_user_config = args.config_file is not None
use_user_config = args.dataset_config is not None
if args.seed is not None:
set_seed(args.seed)
@@ -64,8 +64,8 @@ def train(args):
# データセットを準備する
blueprint_generator = BlueprintGenerator(ConfigSanitizer(True, True, True))
if use_user_config:
print(f"Load config file from {args.config_file}")
user_config = config_util.load_user_config(args.config_file)
print(f"Load dataset config from {args.dataset_config}")
user_config = config_util.load_user_config(args.dataset_config)
ignored = ["train_data_dir", "reg_data_dir", "in_json"]
if any(getattr(args, attr) is not None for attr in ignored):
print(