revert a change

facebookincubator · Jun 25, 2023 · cede684 · cede684
1 parent a4b1e38
commit cede684
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
@@ -257,7 +257,7 @@ def attention(self, x):
                 )
             return out
 
-    def forward(self, *args, seqlens=None):
+    def forward(self, *args):
         """forward pass for calling mha module"""
         assert len(args) >= 1
         x = args[0]