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

fix to work ControlNetSubset with custom_attributes

Browse Source
This commit is contained in:
Kohya S
2024-10-20 08:45:27 +09:00
parent aa932429d1
commit 7fe8e162cb
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -578,6 +578,7 @@ class ControlNetSubset(BaseSubset):
caption_suffix, caption_suffix,
token_warmup_min, token_warmup_min,
token_warmup_step, token_warmup_step,
custom_attributes: Optional[Dict[str, Any]] = None,
) -> None: ) -> None:
assert image_dir is not None, "image_dir must be specified / image_dirは指定が必須です" assert image_dir is not None, "image_dir must be specified / image_dirは指定が必須です"
@@ -602,6 +603,7 @@ class ControlNetSubset(BaseSubset):
caption_suffix, caption_suffix,
token_warmup_min, token_warmup_min,
token_warmup_step, token_warmup_step,
custom_attributes=custom_attributes,
) )
self.conditioning_data_dir = conditioning_data_dir self.conditioning_data_dir = conditioning_data_dir