add nightly marker for torch.compile test

huggingface · Dec 9, 2024 · bc874fc · bc874fc
1 parent e9fccb6
commit bc874fc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/quantization/torchao/test_torchao.py b/tests/quantization/torchao/test_torchao.py
@@ -365,6 +365,7 @@ def test_training(self):
                 self.assertTrue(module.adapter[1].weight.grad is not None)
                 self.assertTrue(module.adapter[1].weight.grad.norm().item() > 0)
 
+    @nightly
     def test_torch_compile(self):
         r"""Test that verifies if torch.compile works with torchao quantization."""
         quantization_config = TorchAoConfig("int8_weight_only")