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

Signed-off-by: Lin, Fanli <fanli.lin@intel.com>
2025-11-11 23:29:33 +08:00
parent a90ad7d838
commit b886068056
1 changed files with 1 additions and 1 deletions
--- a/vllm/model_executor/models/pixtral.py
+++ b/vllm/model_executor/models/pixtral.py
@@ -1109,7 +1109,7 @@ class PixtralHFAttention(nn.Module):
            )
            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)

        return attn_output, None