xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 06:15:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix] Fix RuntimeError in PixtralHFAttention on CPU/XPU (#28444)

Browse Source 
Signed-off-by: Lin, Fanli <fanli.lin@intel.com>
This commit is contained in:
Fanli Lin 2025-11-11 23:29:33 +08:00 committed by GitHub
parent a90ad7d838
commit b886068056
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/model_executor/models/pixtral.py
View File

@ -1109,7 +1109,7 @@ class PixtralHFAttention(nn.Module):
) )
out = out.transpose(1, 2) out = out.transpose(1, 2)
out = out.view(batch, patches, self.n_heads * self.head_dim) out = out.reshape(batch, patches, self.n_heads * self.head_dim)
attn_output, _ = self.o_proj(out) attn_output, _ = self.o_proj(out)
return attn_output, None return attn_output, None