From a7e034c543d145cd54bc46290ebab9844c2de73a Mon Sep 17 00:00:00 2001 From: tbsbngrtz Date: Fri, 2 Oct 2026 14:29:23 +0200 Subject: [PATCH] Tag Krea 2 text encoder outputs as denoiser_input_fields --- src/diffusers/modular_pipelines/krea2/encoders.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/diffusers/modular_pipelines/krea2/encoders.py b/src/diffusers/modular_pipelines/krea2/encoders.py index a07f75305557..154f1d905ade 100644 --- a/src/diffusers/modular_pipelines/krea2/encoders.py +++ b/src/diffusers/modular_pipelines/krea2/encoders.py @@ -110,21 +110,25 @@ def intermediate_outputs(self) -> list[OutputParam]: OutputParam( name="prompt_embeds", type_hint=torch.Tensor, + kwargs_type="denoiser_input_fields", description="Per-prompt stacked text features (B, text_seq_len, num_text_layers, text_hidden_dim).", ), OutputParam( name="prompt_embeds_mask", type_hint=torch.Tensor, + kwargs_type="denoiser_input_fields", description="Per-prompt boolean text mask (B, text_seq_len).", ), OutputParam( name="negative_prompt_embeds", type_hint=torch.Tensor, + kwargs_type="denoiser_input_fields", description="Per-prompt negative text features (only when guidance is enabled).", ), OutputParam( name="negative_prompt_embeds_mask", type_hint=torch.Tensor, + kwargs_type="denoiser_input_fields", description="Per-prompt negative text mask (only when guidance is enabled).", ), ] @@ -254,11 +258,13 @@ def intermediate_outputs(self) -> list[OutputParam]: OutputParam( name="prompt_embeds", type_hint=torch.Tensor, + kwargs_type="denoiser_input_fields", description="Per-prompt stacked text features (B, text_seq_len, num_text_layers, text_hidden_dim).", ), OutputParam( name="prompt_embeds_mask", type_hint=torch.Tensor, + kwargs_type="denoiser_input_fields", description="Per-prompt boolean text mask (B, text_seq_len).", ), ]