Add require_read_token to fp8 tests (#36189)

fix
huggingface · Feb 14, 2025 · cb586a3 · cb586a3
1 parent 5f726f8
commit cb586a3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/quantization/finegrained_fp8/test_fp8.py b/tests/quantization/finegrained_fp8/test_fp8.py
@@ -20,6 +20,7 @@
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, FineGrainedFP8Config, OPTForCausalLM
 from transformers.testing_utils import (
     require_accelerate,
+    require_read_token,
     require_torch_gpu,
     require_torch_multi_gpu,
     slow,
@@ -59,6 +60,7 @@ def test_from_dict(self):
 
 @slow
 @require_accelerate
+@require_read_token
 @require_torch_gpu
 class FP8QuantizerTest(unittest.TestCase):
     model_name = "meta-llama/Llama-3.2-1B"