Upload folder using huggingface_hub
Browse files- modeling_chexagent.py +2 -2
modeling_chexagent.py
CHANGED
|
@@ -44,8 +44,8 @@ from .modeling_visual import CLIPModel
|
|
| 44 |
from .tokenization_chexagent import CheXagentTokenizer
|
| 45 |
|
| 46 |
try:
|
| 47 |
-
from flash_attn import flash_attn_func, flash_attn_varlen_func
|
| 48 |
-
from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input #
|
| 49 |
except:
|
| 50 |
pass
|
| 51 |
|
|
|
|
| 44 |
from .tokenization_chexagent import CheXagentTokenizer
|
| 45 |
|
| 46 |
try:
|
| 47 |
+
from flash_attn import flash_attn_func, flash_attn_varlen_func # type: ignore
|
| 48 |
+
from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input # type: ignore
|
| 49 |
except:
|
| 50 |
pass
|
| 51 |
|