Tag Krea 2 text encoder outputs as denoiser_input_fields (#14925)

Author: tbsbngrtz

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).",
             ),
         ]