fix(nodes): controlnet input accepts list or single controlnet

This commit is contained in:
psychedelicious 2023-05-19 16:10:39 +10:00 committed by Kent Keirsey
parent 009407ea3f
commit be79d088c0
2 changed files with 2 additions and 3 deletions

View File

@ -110,7 +110,7 @@ class ControlOutput(BaseInvocationOutput):
"""node output for ControlNet info"""
# fmt: off
type: Literal["control_output"] = "control_output"
control: Optional[ControlField] = Field(default=None, description="The control info dict")
control: ControlField = Field(default=None, description="The control info dict")
# fmt: on

View File

@ -173,8 +173,7 @@ class TextToLatentsInvocation(BaseInvocation):
seamless: bool = Field(default=False, description="Whether or not to generate an image that can tile without seams", )
seamless_axes: str = Field(default="", description="The axes to tile the image on, 'x' and/or 'y'")
progress_images: bool = Field(default=False, description="Whether or not to produce progress images during generation", )
control: list[ControlField] = Field(default=None, description="The controlnet(s) to use")
# control: ControlField | List[ControlField] = Field(default=None, description="The controlnet(s) to use")
control: Union[ControlField, List[ControlField]] = Field(default=None, description="The controlnet(s) to use")
# fmt: on
# Schema customisation