disable flash attn 2 (#1065)

ModelCloud · Jan 9, 2025 · 0c6452b · 0c6452b
1 parent 6984fde
commit 0c6452b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/test_internlm.py b/tests/models/test_internlm.py
@@ -22,6 +22,7 @@ class TestInternlm(ModelTest):
     NATIVE_ARC_CHALLENGE_ACC_NORM = 0.4309
     TRUST_REMOTE_CODE = True
     USE_VLLM = False
+    DISABLE_FLASH_ATTN = True
 
     def test_internlm(self):
         # transformers<=4.44.2 run normal