Skip torchao tests on XPU

huggingface · Sep 19, 2024 · 134b50b · 134b50b
1 parent 0d4eaf0
commit 134b50b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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):