Add batch_keyframe_idx on ControlNetUnit (#2926)

batch_keyframe_idx is needed by AnimateDiff for SparseCtrl
2024-05-23 05:54:13 -07:00 · 2024-05-23 05:54:13 -07:00 · ba3984a022
parent 92e4b12a73
commit ba3984a022
1 changed files with 1 additions and 0 deletions
--- a/internal_controlnet/args.py
+++ b/internal_controlnet/args.py
@ -228,6 +228,7 @@ class ControlNetUnit(BaseModel):
    animatediff_batch: bool = False
    batch_modifiers: list = []
    batch_image_files: list = []
+    batch_keyframe_idx: Optional[str|list] = None

    @property
    def accepts_multiple_inputs(self) -> bool: