Update modeling_orion.py

modeling_orion.py

@@ -30,9 +30,12 @@ from transformers.utils import (
 )
 import logging
 if is_flash_attn_2_available():
-    from flash_attn import flash_attn_func, flash_attn_varlen_func
-    from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input  # noqa
-
+    try:
+        from flash_attn import flash_attn_func, flash_attn_varlen_func
+        from flash_attn.bert_padding import index_first_axis, pad_input, unpad_input  # noqa
+    except ImportError:
+        pass
+        
 logger = logging.getLogger(__name__)
 
 _CONFIG_FOR_DOC = "OrionConfig"