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