[Bugfix][ROCm] Fix ViT rotary embeddings for torch.compile compatibility on ROCm (#27748)

Signed-off-by: vllmellm <vllm.ellm@embeddedllm.com>
This commit is contained in:
vllmellm
2025-11-04 09:12:19 +08:00
committed by GitHub
parent 7956b0c0bc
commit b13a447546
2 changed files with 8 additions and 5 deletions

View File

@@ -370,7 +370,7 @@ class Glm4vVisionAttention(nn.Module):
cu_seqlens_k=cu_seqlens,
max_seqlen_q=max_seqlen,
max_seqlen_k=max_seqlen,
dropout_p=0,
dropout_p=0.0,
causal=False,
)