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

Fix parameters

Browse Source
This commit is contained in:
rockerBOO
2025-01-28 13:45:07 -05:00
parent 0da9e10fc5
commit 46414bb689
2 changed files with 26 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
library/train_util.py
View File

@@ -445,11 +445,11 @@ class BaseSubset:
custom_attributes: Optional[Dict[str, Any]] = None,
validation_seed: Optional[int] = None,
validation_split: Optional[float] = 0.0,
preference: bool,
preference_caption_prefix: Optional[str],
preference_caption_suffix: Optional[str],
non_preference_caption_prefix: Optional[str],
non_preference_caption_suffix: Optional[str],
preference: bool = False,
preference_caption_prefix: Optional[str] = None,
preference_caption_suffix: Optional[str] = None,
non_preference_caption_prefix: Optional[str] = None,
non_preference_caption_suffix: Optional[str] = None,
) -> None:
self.image_dir = image_dir
self.alpha_mask = alpha_mask if alpha_mask is not None else False
@@ -516,11 +516,11 @@ class DreamBoothSubset(BaseSubset):
custom_attributes: Optional[Dict[str, Any]] = None,
validation_seed: Optional[int] = None,
validation_split: Optional[float] = 0.0,
preference: bool,
preference_caption_prefix,
preference_caption_suffix,
non_preference_caption_prefix,
non_preference_caption_suffix,
preference: bool = False,
preference_caption_prefix: Optional[str] = None,
preference_caption_suffix: Optional[str] = None,
non_preference_caption_prefix: Optional[str] = None,
non_preference_caption_suffix: Optional[str] = None,
) -> None:
assert image_dir is not None, "image_dir must be specified / image_dirは指定が必須です"
@@ -548,11 +548,11 @@ class DreamBoothSubset(BaseSubset):
custom_attributes=custom_attributes,
validation_seed=validation_seed,
validation_split=validation_split,
preference,
preference_caption_prefix,
preference_caption_suffix,
non_preference_caption_prefix,
non_preference_caption_suffix,
preference=preference,
preference_caption_prefix=preference_caption_prefix,
preference_caption_suffix=preference_caption_suffix,
non_preference_caption_prefix=non_preference_caption_prefix,
non_preference_caption_suffix=non_preference_caption_suffix,
)
self.is_reg = is_reg
@@ -660,11 +660,11 @@ class ControlNetSubset(BaseSubset):
custom_attributes: Optional[Dict[str, Any]] = None,
validation_seed: Optional[int] = None,
validation_split: Optional[float] = 0.0,
preference,
preference_caption_prefix,
preference_caption_suffix,
non_preference_caption_prefix,
non_preference_caption_suffix
preference: bool = False,
preference_caption_prefix: Optional[str] = None,
preference_caption_suffix: Optional[str] = None,
non_preference_caption_prefix: Optional[str] = None,
non_preference_caption_suffix: Optional[str] = None,
) -> None:
assert image_dir is not None, "image_dir must be specified / image_dirは指定が必須です"
@@ -692,11 +692,11 @@ class ControlNetSubset(BaseSubset):
custom_attributes=custom_attributes,
validation_seed=validation_seed,
validation_split=validation_split,
preference,
preference_caption_prefix,
preference_caption_suffix,
non_preference_caption_prefix,
non_preference_caption_suffix,
preference=preference,
preference_caption_prefix=preference_caption_prefix,
preference_caption_suffix=preference_caption_suffix,
non_preference_caption_prefix=non_preference_caption_prefix,
non_preference_caption_suffix=non_preference_caption_suffix,
)
self.conditioning_data_dir = conditioning_data_dir

2
requirements.txt
View File

@@ -20,7 +20,7 @@ voluptuous==0.13.1
huggingface-hub==0.24.5
# for Image utils
imagesize==1.4.1
numpy<=2.0
numpy<2.0
# for BLIP captioning
# requests==2.28.2
# timm==0.6.12