From e8717d3bef74a4cb9ce8e0cbda2d1da78e64c4bb Mon Sep 17 00:00:00 2001
From: NanoCode012 <kevinvong@rocketmail.com>
Date: Mon, 29 May 2023 10:08:30 +0900
Subject: [PATCH] Remove disable

---
 src/axolotl/flash_attn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/axolotl/flash_attn.py b/src/axolotl/flash_attn.py
index d532e15a8..c7bd12c66 100644
--- a/src/axolotl/flash_attn.py
+++ b/src/axolotl/flash_attn.py
@@ -15,7 +15,7 @@ from flash_attn.flash_attn_interface import flash_attn_unpadded_qkvpacked_func
 from flash_attn.bert_padding import unpad_input, pad_input
 
 
-def forward(  # pylint: disable=too-many-arguments
+def forward(
     self,
     hidden_states: torch.Tensor,
     attention_mask: Optional[torch.Tensor] = None,