ci for fa3

2025-05-18 00:49:15 -07:00
parent fb5ef6d445
commit a064f1c9b4
6 changed files with 46 additions and 9 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -458,6 +458,7 @@ def cleanup_monkeypatches():
        ("transformers.trainer",),
        ("transformers", ["Trainer"]),
        ("transformers.loss.loss_utils",),
+        ("transformers.modeling_flash_attention_utils",),
    ]
    for module_name_tuple in modules_to_reset:
        module_name = module_name_tuple[0]