Add ControlNetUnit.batch_mask_dir for compatibility (#2865)

Mikubill · May 7, 2024 · 8d1389d · 8d1389d
1 parent 084dd01
commit 8d1389d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal_controlnet/args.py b/internal_controlnet/args.py
@@ -221,6 +221,9 @@ def parse_ipadapter_input(cls, value) -> Optional[List[torch.Tensor]]:
     # 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."""