Add animatediff_batch flag on ControlNetUnit (#2874)

Mikubill · May 10, 2024 · ae9528b · ae9528b
1 parent aa0a72a
commit ae9528b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/internal_controlnet/args.py b/internal_controlnet/args.py
@@ -221,8 +221,10 @@ 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.
+    # AnimateDiff compatibility fields.
+    # TODO: Find a better way in AnimateDiff to deal with these extra fields.
     batch_mask_dir: Optional[str] = None
+    animatediff_batch: bool = False
 
     @property
     def accepts_multiple_inputs(self) -> bool: