diff --git a/invokeai/app/invocations/compel.py b/invokeai/app/invocations/compel.py index fd8c0ea16b..fa1b6939d2 100644 --- a/invokeai/app/invocations/compel.py +++ b/invokeai/app/invocations/compel.py @@ -331,8 +331,8 @@ class SDXLCompelPromptInvocation(BaseInvocation, SDXLPromptInvocationBase): crop_left: int = Field(0, description="") target_width: int = Field(1024, description="") target_height: int = Field(1024, description="") - clip1: ClipField = Field(None, description="Clip to use") - clip2: ClipField = Field(None, description="Clip to use") + clip: ClipField = Field(None, description="Clip to use") + clip2: ClipField = Field(None, description="Clip2 to use") # Schema customisation class Config(InvocationConfig): @@ -348,7 +348,7 @@ class SDXLCompelPromptInvocation(BaseInvocation, SDXLPromptInvocationBase): @torch.no_grad() def invoke(self, context: InvocationContext) -> CompelOutput: - c1, c1_pooled, ec1 = self.run_clip_compel(context, self.clip1, self.prompt, False) + c1, c1_pooled, ec1 = self.run_clip_compel(context, self.clip, self.prompt, False) if self.style.strip() == "": c2, c2_pooled, ec2 = self.run_clip_compel(context, self.clip2, self.prompt, True) else: @@ -451,8 +451,8 @@ class SDXLRawPromptInvocation(BaseInvocation, SDXLPromptInvocationBase): crop_left: int = Field(0, description="") target_width: int = Field(1024, description="") target_height: int = Field(1024, description="") - clip1: ClipField = Field(None, description="Clip to use") - clip2: ClipField = Field(None, description="Clip to use") + clip: ClipField = Field(None, description="Clip to use") + clip2: ClipField = Field(None, description="Clip2 to use") # Schema customisation class Config(InvocationConfig): @@ -468,7 +468,7 @@ class SDXLRawPromptInvocation(BaseInvocation, SDXLPromptInvocationBase): @torch.no_grad() def invoke(self, context: InvocationContext) -> CompelOutput: - c1, c1_pooled, ec1 = self.run_clip_raw(context, self.clip1, self.prompt, False) + c1, c1_pooled, ec1 = self.run_clip_raw(context, self.clip, self.prompt, False) if self.style.strip() == "": c2, c2_pooled, ec2 = self.run_clip_raw(context, self.clip2, self.prompt, True) else: