From f2db1e8c6dfd9db9473fbf2719df86a2f3f9c34c Mon Sep 17 00:00:00 2001 From: fsx950223 Date: Mon, 26 Jun 2023 12:08:25 +0800 Subject: [PATCH] add end line --- python/aitemplate/frontend/nn/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/aitemplate/frontend/nn/attention.py b/python/aitemplate/frontend/nn/attention.py index 35d9b483e..1f1240762 100644 --- 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 \ No newline at end of file + return attn