Add more animatediff compatible fields (#2878)

pull/2881/head
Chenlei Hu 2024-05-10 14:21:07 -04:00 committed by GitHub
parent 85c731070f
commit b30f0ac4a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions

View File

@ -226,6 +226,8 @@ class ControlNetUnit(BaseModel):
# TODO: Find a better way in AnimateDiff to deal with these extra fields.
batch_mask_dir: Optional[str] = None
animatediff_batch: bool = False
batch_modifiers: list = []
batch_image_files: list = []
@property
def accepts_multiple_inputs(self) -> bool: