Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added an additional "raw_processed_image" output port to controlnets, mainly so could route ImageField to a ShowImage node

Browse Source
This commit is contained in:
user1 2023-05-04 23:14:14 -07:00 committed by Kent Keirsey
parent 21251ce12c
commit 754017b59e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/invocations/controlnet_image_processors.py
View File

@ -47,7 +47,7 @@ class ControlOutput(BaseInvocationOutput):
# fmt: off # fmt: off
type: Literal["control_output"] = "control_output" type: Literal["control_output"] = "control_output"
control: Optional[ControlField] = Field(default=None, description="The control info dict") control: Optional[ControlField] = Field(default=None, description="The control info dict")
# image: ImageField = Field(default=None, description="outputs just them image info (which is also included in control output)") raw_processed_image: ImageField = Field(default=None, description="outputs just them image info (which is also included in control output)")
# fmt: on # fmt: on
@ -101,7 +101,8 @@ class PreprocessedControlInvocation(BaseInvocation, PILInvocationConfig):
image=image_field, image=image_field,
control_model=self.control_model, control_model=self.control_model,
control_weight=self.control_weight, control_weight=self.control_weight,
) ),
raw_processed_image=image_field,
) )