From 5ae9b2cac0b5b015e41eae328df56b97e8205262 Mon Sep 17 00:00:00 2001 From: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com> Date: Thu, 10 Apr 2025 14:57:15 +0200 Subject: [PATCH] Quark Quantization gated repo (#37412) * fix * empty commit * empty * nit * fix maybe ? --- tests/quantization/quark_integration/test_quark.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/quantization/quark_integration/test_quark.py b/tests/quantization/quark_integration/test_quark.py index 728ca36482..5a725bff32 100644 --- a/tests/quantization/quark_integration/test_quark.py +++ b/tests/quantization/quark_integration/test_quark.py @@ -19,6 +19,7 @@ from transformers.testing_utils import ( is_torch_available, require_accelerate, require_quark, + require_read_token, require_torch_gpu, require_torch_multi_gpu, slow, @@ -56,6 +57,7 @@ class QuarkTest(unittest.TestCase): EXPECTED_RELATIVE_DIFFERENCE = 1.66 device_map = None + @require_read_token @classmethod def setUpClass(cls): """