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

faaany · web-flow · commit b886068056a0 · 2025-11-11T15:29:33.000Z
Signed-off-by: Lin, Fanli &lt;fanli.lin@intel.com&gt;
diff --git a/vllm/model_executor/models/pixtral.py b/vllm/model_executor/models/pixtral.py
@@ -1109,7 +1109,7 @@ def forward(
             )
             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

Original file line number	Diff line number	Diff line change
`@@ -1109,7 +1109,7 @@ def forward(`
`1109`	`1109`	`)`
`1110`	`1110`	`out = out.transpose(1, 2)`
`1111`	`1111`
`1112`		`- out = out.view(batch, patches, self.n_heads * self.head_dim)`
	`1112`	`+ out = out.reshape(batch, patches, self.n_heads * self.head_dim)`
`1113`	`1113`	`attn_output, _ = self.o_proj(out)`
`1114`	`1114`
`1115`	`1115`	`return attn_output, None`