add end line

facebookincubator · Jun 26, 2023 · f2db1e8 · f2db1e8
1 parent 8cc9e76
commit f2db1e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/aitemplate/frontend/nn/attention.py b/python/aitemplate/frontend/nn/attention.py
@@ -405,4 +405,4 @@ def __init__(self) -> None:
 
     def forward(self, q, k, v):
         attn = ops.mem_eff_attention(causal=False)(q, k, v)
-        return attn
+        return attn