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

Cleanup order, types, print to logger

Browse Source
This commit is contained in:
rockerBOO
2025-01-03 01:26:45 -05:00
parent 534059dea5
commit c8c3569df2
2 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
library/config_util.py
View File

@@ -485,7 +485,7 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
dataset = dataset_klass(subsets=subsets, **asdict(dataset_blueprint.params))
datasets.append(dataset)
val_datasets:List[Union[DreamBoothDataset, FineTuningDataset, ControlNetDataset]] = []
val_datasets: List[Union[DreamBoothDataset, FineTuningDataset, ControlNetDataset]] = []
for dataset_blueprint in dataset_group_blueprint.datasets:
if dataset_blueprint.params.validation_split <= 0.0:
continue
@@ -503,7 +503,6 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
dataset = dataset_klass(subsets=subsets, **asdict(dataset_blueprint.params))
val_datasets.append(dataset)
# print info
def print_info(_datasets):
info = ""
for i, dataset in enumerate(_datasets):
@@ -565,7 +564,7 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
print_info(datasets)
if len(val_datasets) > 0:
print("Validation dataset")
logger.info("Validation dataset")
print_info(val_datasets)
if len(val_datasets) > 0:
@@ -610,7 +609,7 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
" ",
)
logger.info(f"{info}")
logger.info(info)
# make buckets first because it determines the length of dataset
# and set the same seed for all datasets