From 754017b59e25fc85e04912ebe451e82142f0099e Mon Sep 17 00:00:00 2001 From: user1 Date: Thu, 4 May 2023 23:14:14 -0700 Subject: [PATCH] Added an additional "raw_processed_image" output port to controlnets, mainly so could route ImageField to a ShowImage node --- invokeai/app/invocations/controlnet_image_processors.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/invokeai/app/invocations/controlnet_image_processors.py b/invokeai/app/invocations/controlnet_image_processors.py index 60d1fabb21..e5f0d930b2 100644 --- a/invokeai/app/invocations/controlnet_image_processors.py +++ b/invokeai/app/invocations/controlnet_image_processors.py @@ -47,7 +47,7 @@ class ControlOutput(BaseInvocationOutput): # fmt: off type: Literal["control_output"] = "control_output" 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 @@ -101,7 +101,8 @@ class PreprocessedControlInvocation(BaseInvocation, PILInvocationConfig): image=image_field, control_model=self.control_model, control_weight=self.control_weight, - ) + ), + raw_processed_image=image_field, )