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

Update config_util.py

Browse Source
This commit is contained in:
gesen2egee
2024-04-11 01:52:14 +08:00
committed by GitHub
parent 3b251b758d
commit 459b12539b
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
library/config_util.py
View File

@@ -512,8 +512,7 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
subset_blueprint.params.token_warmup_step = 0 subset_blueprint.params.token_warmup_step = 0
if subset_klass != DreamBoothSubset or (subset_klass == DreamBoothSubset and not subset_blueprint.params.is_reg): if subset_klass != DreamBoothSubset or (subset_klass == DreamBoothSubset and not subset_blueprint.params.is_reg):
subset = subset_klass(**asdict(subset_blueprint.params)) subsets.append(subset_klass(**asdict(subset_blueprint.params)))
subsets.append(subset)
dataset = dataset_klass(subsets=subsets, is_train=False, **asdict(dataset_blueprint.params)) dataset = dataset_klass(subsets=subsets, is_train=False, **asdict(dataset_blueprint.params))
val_datasets.append(dataset) val_datasets.append(dataset)