SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Quark Quantization gated repo (#37412)

Browse Source 
* fix

* empty commit

* empty

* nit

* fix maybe ?
This commit is contained in:
Mohamed Mekkouri
2025-04-10 14:57:15 +02:00
committed by GitHub
parent d9e76656ae
commit 5ae9b2cac0
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/quantization/quark_integration/test_quark.py
View File

@@ -19,6 +19,7 @@ from transformers.testing_utils import (
is_torch_available, is_torch_available,
require_accelerate, require_accelerate,
require_quark, require_quark,
require_read_token,
require_torch_gpu, require_torch_gpu,
require_torch_multi_gpu, require_torch_multi_gpu,
slow, slow,
@@ -56,6 +57,7 @@ class QuarkTest(unittest.TestCase):
EXPECTED_RELATIVE_DIFFERENCE = 1.66 EXPECTED_RELATIVE_DIFFERENCE = 1.66
device_map = None device_map = None
@require_read_token
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
""" """