tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

add transformers module to cleanup

Browse Source
This commit is contained in:
Wing Lian
2024-12-08 14:52:54 -05:00
parent 26bd81cec0
commit 39ab9626f1
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/conftest.py
View File

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