Add ControlNetUnit.batch_mask_dir for compatibility (#2865)

pull/2873/head
Chenlei Hu 2024-05-07 17:58:48 -04:00 committed by GitHub
parent 084dd0107b
commit 8d1389da5e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions

View File

@ -221,6 +221,9 @@ class ControlNetUnit(BaseModel):
# The mask to be used on top of the image.
mask: Optional[Any] = None
# Backward compatible with animatediff impl.
batch_mask_dir: Optional[str] = None
@property
def accepts_multiple_inputs(self) -> bool:
"""This unit can accept multiple input images."""