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

Rename min_orig_resolution to skip_image_resolution; remove max_orig_resolution

Browse Source
This commit is contained in:
woctordho
2026-02-20 22:59:40 +08:00
parent af3a55b21d
commit 47afa8b2bd
3 changed files with 22 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
library/config_util.py
View File

@@ -108,8 +108,7 @@ class BaseDatasetParams:
validation_seed: Optional[int] = None
validation_split: float = 0.0
resize_interpolation: Optional[str] = None
min_orig_resolution: float = 0.0
max_orig_resolution: float = float("inf")
skip_image_resolution: float = 0.0
@dataclass
class DreamBoothDatasetParams(BaseDatasetParams):
@@ -246,8 +245,7 @@ class ConfigSanitizer:
"resolution": functools.partial(__validate_and_convert_scalar_or_twodim.__func__, int),
"network_multiplier": float,
"resize_interpolation": str,
"min_orig_resolution": Any(float, int),
"max_orig_resolution": Any(float, int),
"skip_image_resolution": Any(float, int),
}
# options handled by argparse but not handled by user config
@@ -260,8 +258,7 @@ class ConfigSanitizer:
ARGPARSE_NULLABLE_OPTNAMES = [
"face_crop_aug_range",
"resolution",
"min_orig_resolution",
"max_orig_resolution",
"skip_image_resolution",
]
# prepare map because option name may differ among argparse and user config
ARGPARSE_OPTNAME_TO_CONFIG_OPTNAME = {
@@ -534,8 +531,7 @@ def generate_dataset_group_by_blueprint(dataset_group_blueprint: DatasetGroupBlu
[{dataset_type} {i}]
batch_size: {dataset.batch_size}
resolution: {(dataset.width, dataset.height)}
min_orig_resolution: {dataset.min_orig_resolution}
max_orig_resolution: {dataset.max_orig_resolution}
skip_image_resolution: {dataset.skip_image_resolution}
resize_interpolation: {dataset.resize_interpolation}
enable_bucket: {dataset.enable_bucket}
""")