Rename _supports_flash_attn_2 in examples and tests (#39471)

* delete `_supports_flash_attn_2` from examples and tests * simplify docs
2025-07-21 14:02:57 +02:00
parent 3a152e3a5c
commit 8c102e2eb1
22 changed files with 39 additions and 58 deletions
--- a/tests/generation/test_utils.py
+++ b/tests/generation/test_utils.py
@@ -2297,8 +2297,8 @@ class GenerationTesterMixin:
        max_new_tokens = 3
        support_flag = {
            "sdpa": "_supports_sdpa",
-            "flash_attention_2": "_supports_flash_attn_2",
-            "flash_attention_3": "_supports_flash_attn_3",
+            "flash_attention_2": "_supports_flash_attn",
+            "flash_attention_3": "_supports_flash_attn",
        }

        set_model_tester_for_less_flaky_test(self)