radna
/

Triton-InternViT-6B-448px-V1-5

Image Feature Extraction

feature-extraction

Model card Files Files and versions Community

radna commited on Jun 14

Commit

6e45b59

•

1 Parent(s): c808872

Update flash_attention.py

Files changed (1) hide show

flash_attention.py +2 -2

flash_attention.py CHANGED Viewed

@@ -3,8 +3,8 @@ import torch.nn as nn
 from einops import rearrange
 try:
-    from triton_flash_atn import _attention
-    from triton_bert_pading import pad_input, unpad_input
 except:
     print("FlashAttention is not installed.")

 from einops import rearrange
 try:
+    from .triton_flash_atn import _attention
+    from .triton_bert_pading import pad_input, unpad_input
 except:
     print("FlashAttention is not installed.")