From 39ab9626f109c0048fa133c4dbe764d0b35f75cd Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 8 Dec 2024 14:52:54 -0500 Subject: [PATCH] add transformers module to cleanup --- tests/conftest.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/conftest.py b/tests/conftest.py index a9dde9dd8..030ddd68d 100644 --- 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",),