Expectations re-order and corrected FA3 skip (#39195)

* Fix Expectations and a FA3 skip * Fixed docstring * Added context for Default expectation
2025-07-07 11:42:33 +02:00
parent b0a8e0b8d7
commit a325409a50
2 changed files with 16 additions and 8 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -4306,7 +4306,7 @@ class ModelTesterMixin:
    def test_flash_attn_2_from_config(self):
        self.flash_attn_from_config(attn_implementation="flash_attention_2")

-    @require_flash_attn
+    @require_flash_attn_3
    @require_torch_gpu
    @mark.flash_attn_3_test
    @slow