nclgbd commited on
Commit
1018e93
·
verified ·
1 Parent(s): fad8e18

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 # noqa
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