diff --git a/tests/test_gpu_examples.py b/tests/test_gpu_examples.py index 91b09a0fbd..f2c2ae63d7 100644 --- a/tests/test_gpu_examples.py +++ b/tests/test_gpu_examples.py @@ -1930,8 +1930,10 @@ def test_causal_lm_training_aqlm(self): 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