add transformers module to cleanup

2024-12-08 14:52:54 -05:00
parent 26bd81cec0
commit 39ab9626f1
1 changed files with 1 additions and 0 deletions
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -138,6 +138,7 @@ def cleanup_monkeypatches():

    # Reset other known monkeypatches
    modules_to_reset: list[tuple[str, list[str]]] = [
+        ("transformers",),
        ("transformers.models.llama.modeling_llama", ["LlamaFlashAttention2"]),
        ("transformers.trainer", ["Trainer"]),
        ("transformers.loss.loss_utils",),