[Bug] Fix usage of .transpose() and .view() consecutively. (#11979)

2025-01-13 14:24:10 +08:00
parent f7b3ba82c3
commit 9dd02d85ca
2 changed files with 2 additions and 2 deletions
--- a/vllm/model_executor/models/intern_vit.py
+++ b/vllm/model_executor/models/intern_vit.py
@@ -271,7 +271,7 @@ class InternSdpaAttention(nn.Module):
        v = v.transpose(1, 2)

        x = F.scaled_dot_product_attention(q, k, v, scale=self.scale)
-        x = x.transpose(1, 2).view(B, N, -1)
+        x = x.transpose(1, 2).reshape(B, N, -1)

        x = self.proj(x)
        return x