Update app.py

app.py

@@ -216,7 +216,7 @@ def attention(q: Tensor, k: Tensor, v: Tensor, pe: Tensor) -> Tensor:
 
     x = torch.nn.functional.scaled_dot_product_attention(q, k, v)
     # x = rearrange(x, "B H L D -> B L (H D)")
-    x = x.permute(0, 2, 1, 3).contiguous().reshape(x.size(0), x.size(2), -1)
+    x = x.permute(0, 2, 1, 3).reshape(x.size(0), x.size(2), -1)
 
     return x