diff --git a/tests/test_gpu_examples.py b/tests/test_gpu_examples.py index 781ba7b9a8..6880299755 100644 --- a/tests/test_gpu_examples.py +++ b/tests/test_gpu_examples.py @@ -3013,6 +3013,7 @@ def test_causal_lm_training_multi_gpu_eetq(self): assert trainer.state.log_history[-1]["train_loss"] is not None +@require_non_xpu @require_torch_gpu @require_torchao class PeftTorchaoGPUTests(unittest.TestCase):