We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ecc7597 commit ba28383Copy full SHA for ba28383
models/qwenimage.py
@@ -572,6 +572,7 @@ def forward(
572
encoder_hidden_states_mask: torch.Tensor,
573
temb: torch.Tensor,
574
image_rotary_emb: Optional[Tuple[torch.Tensor, torch.Tensor]] = None,
575
+ transformer_options: dict = {},
576
) -> Tuple[torch.Tensor, torch.Tensor]:
577
"""
578
Forward pass for the transformer block.
@@ -625,6 +626,7 @@ def forward(
625
626
encoder_hidden_states=txt_modulated, # Text stream ("context")
627
encoder_hidden_states_mask=encoder_hidden_states_mask,
628
image_rotary_emb=image_rotary_emb,
629
+ transformer_options=transformer_options,
630
)
631
632
# QwenAttnProcessor2_0 returns (img_output, txt_output) when encoder_hidden_states is provided
0 commit comments