Spaces:

Dovakiins
/

qwerrwe

Build error

Nanobit commited on May 29, 2023

Commit

e8717d3

1 Parent(s): 54c3b5b

Remove disable

Files changed (1) hide show

src/axolotl/flash_attn.py CHANGED Viewed

@@ -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
     self,
     hidden_states: torch.Tensor,
     attention_mask: Optional[torch.Tensor] = None,

 from flash_attn.bert_padding import unpad_input, pad_input
+def forward(
     self,
     hidden_states: torch.Tensor,
     attention_mask: Optional[torch.Tensor] = None,