fix to work ControlNetSubset with custom_attributes

2026-04-06 21:52:27 +00:00 · 2024-10-20 08:45:27 +09:00
parent aa932429d1
commit 7fe8e162cb
1 changed files with 2 additions and 0 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -578,6 +578,7 @@ class ControlNetSubset(BaseSubset):
        caption_suffix,
        token_warmup_min,
        token_warmup_step,
+        custom_attributes: Optional[Dict[str, Any]] = None,
    ) -> None:
        assert image_dir is not None, "image_dir must be specified / image_dirは指定が必須です"

@@ -602,6 +603,7 @@ class ControlNetSubset(BaseSubset):
            caption_suffix,
            token_warmup_min,
            token_warmup_step,
+            custom_attributes=custom_attributes,
        )

        self.conditioning_data_dir = conditioning_data_dir