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:
DKnight54
2025-05-26 00:02:14 +08:00
committed by GitHub
parent a21b6a917e
commit 16584de262
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
library/config_util.py
View File

@@ -576,9 +576,10 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
seed = random.randint(0, 2**31) # actual seed is seed + epoch_no seed = random.randint(0, 2**31) # actual seed is seed + epoch_no
for i, dataset in enumerate(datasets): for i, dataset in enumerate(datasets):
logger.info(f"[Dataset {i}]") logger.info(f"[Dataset {i}]")
dataset.make_buckets()
dataset.set_seed(seed) dataset.set_seed(seed)
dataset.incremental_reg_load()
dataset.make_buckets()
return DatasetGroup(datasets) return DatasetGroup(datasets)