Update modeling_intern_vit.py

modeling_intern_vit.py

@@ -20,9 +20,9 @@ from .configuration_intern_vit import InternVisionConfig
 
 
 try:
-    from triton_flash_atn import _attention
+    from .triton_flash_atn import _attention
 
-    from triton_bert_pading import pad_input, unpad_input
+    from .triton_bert_pading import pad_input, unpad_input
 
     has_flash_attn = True
 except: