radna commited on
Commit
c304902
1 Parent(s): 664bae3

Update modeling_intern_vit.py

Browse files
Files changed (1) hide show
  1. modeling_intern_vit.py +2 -2
modeling_intern_vit.py CHANGED
@@ -20,9 +20,9 @@ from .configuration_intern_vit import InternVisionConfig
20
 
21
 
22
  try:
23
- from triton_flash_atn import _attention
24
 
25
- from triton_bert_pading import pad_input, unpad_input
26
 
27
  has_flash_attn = True
28
  except:
 
20
 
21
 
22
  try:
23
+ from .triton_flash_atn import _attention
24
 
25
+ from .triton_bert_pading import pad_input, unpad_input
26
 
27
  has_flash_attn = True
28
  except: