Update bert_layers.py

commented out triton import to resolve Compilation Issues

bert_layers.py

@@ -25,10 +25,11 @@ from .bert_padding import (index_first_axis,
                                             index_put_first_axis, pad_input,
                                             unpad_input, unpad_input_only)
 
-try:
-    from .flash_attn_triton import flash_attn_qkvpacked_func
-except ImportError as e:
-    flash_attn_qkvpacked_func = None
+# Commented-out to resolve Compilation Issues with triton
+# try:
+#     from .flash_attn_triton import flash_attn_qkvpacked_func
+# except ImportError as e:
+flash_attn_qkvpacked_func = None
 
 logger = logging.getLogger(__name__)