Skip to content

vllm_omni.diffusion.models.ernie_image.fused_rope ¶

Lossless eager Q/K RoPE fusion for ERNIE-Image.

logger `module-attribute` ¶

logger = logging.getLogger(__name__)

try_fused_qk_rotary_emb ¶

try_fused_qk_rotary_emb(
    query: Tensor,
    key: Tensor,
    freqs_cos: Tensor,
    freqs_sin: Tensor,
) -> tuple[Tensor, Tensor] | None

Return fused Q/K outputs, or None to request the eager fallback.