AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-23 09:09:01 +03:00
Code Activity

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

Browse Source
This commit is contained in:
TB
2026-10-02 21:17:50 +02:00
committed by GitHub
parent 578c9b2c66
commit 41566305ce
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/modular_pipelines/krea2/encoders.py
View File

@@ -110,21 +110,25 @@ class Krea2TextEncoderStep(ModularPipelineBlocks):
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 @@ class Krea2TurboTextEncoderStep(Krea2TextEncoderStep):
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).",
),
]