Remove disable

2023-05-29 10:08:30 +09:00
parent 54c3b5b25f
commit e8717d3bef
1 changed files with 1 additions and 1 deletions
--- 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,