fix to work ControlNetSubset with custom_attributes

This commit is contained in:
Kohya S
2024-10-20 08:45:27 +09:00
parent aa932429d1
commit 7fe8e162cb

View File

@@ -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