From 8d1389da5ee9cac4100fbbe54b1b795e22ce00f8 Mon Sep 17 00:00:00 2001 From: Chenlei Hu Date: Tue, 7 May 2024 17:58:48 -0400 Subject: [PATCH] Add ControlNetUnit.batch_mask_dir for compatibility (#2865) --- internal_controlnet/args.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/internal_controlnet/args.py b/internal_controlnet/args.py index f36448e..28ee442 100644 --- a/internal_controlnet/args.py +++ b/internal_controlnet/args.py @@ -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."""