Ensure the attention override returns a contiguous tensor.

2025-09-17 13:17:46 -06:00
parent bb3c5981c7
commit 9bb5c7d3df
1 changed files with 1 additions and 1 deletions
@@ -155,7 +155,7 @@ def attention_bleh(  # noqa: PLR0914
        result = result.transpose(1, 2)
    if not skip_output_reshape:
        result = result.reshape(batch, -1, heads * dim_head)
-    return result
+    return result.contiguous()


 def copy_funattrs(fun, dest=None):