Compare commits

...

2 Commits

Author SHA1 Message Date
b3253faa38 Update tests/test_gpu_examples.py 2024-04-29 18:15:49 +02:00
480d935e72 Update test_gpu_examples.py 2024-04-29 17:43:43 +02:00

View File

@ -1930,8 +1930,10 @@ class PeftAqlmGPUTests(unittest.TestCase):
assert trainer.state.log_history[-1]["train_loss"] is not None
# TODO: unskip the tests once https://github.com/casper-hansen/AutoAWQ/issues/466 is fixed
@require_torch_gpu
@require_auto_awq
@pytest.mark.skip(reason="Needs https://github.com/casper-hansen/AutoAWQ/issues/466 to be fixed first")
class PeftAwqGPUTests(unittest.TestCase):
r"""
Awq + peft tests