fix flash_attn import logic to comply with huggingface import check

modeling_intern_vit.py

@@ -19,20 +19,18 @@ from transformers.utils import logging
 
 from .configuration_intern_vit import InternVisionConfig
 
+has_flash_attn = False
 try:
-    try:  # v1
-        from flash_attn.flash_attn_interface import \
-            flash_attn_unpadded_qkvpacked_func
-    except:  # v2
-        from flash_attn.flash_attn_interface import \
-            flash_attn_varlen_qkvpacked_func as flash_attn_unpadded_qkvpacked_func
-
+    from flash_attn.flash_attn_interface import flash_attn_unpadded_qkvpacked_func
     from flash_attn.bert_padding import pad_input, unpad_input
-
     has_flash_attn = True
-except:
-    print('FlashAttention is not installed.')
-    has_flash_attn = False
+except ImportError:
+    try:
+        from flash_attn.flash_attn_interface import flash_attn_varlen_qkvpacked_func as flash_attn_unpadded_qkvpacked_func
+        from flash_attn.bert_padding import pad_input, unpad_input
+        has_flash_attn = True
+    except ImportError:
+        print('FlashAttention is not installed.')
 
 logger = logging.get_logger(__name__)