broken 🦥 with latest transformers

This commit is contained in:
Wing Lian
2024-12-06 11:34:06 -05:00
parent 84a14fc604
commit 811224d7b7
3 changed files with 16 additions and 1 deletions

View File

@@ -20,6 +20,7 @@ os.environ["WANDB_DISABLED"] = "true"
# pylint: disable=duplicate-code
@pytest.mark.skip(reason="latest unsloth doesn't work with latest transformers")
class TestUnslothQLoRA:
"""
Test class for Unsloth QLoRA Llama models