Fix : HQQ config when hqq not available (#35655)
* fix * make style * adding require_hqq * make style
This commit is contained in:
@@ -19,6 +19,7 @@ import unittest
|
||||
from transformers import AutoModelForCausalLM, AutoTokenizer, HqqConfig
|
||||
from transformers.testing_utils import (
|
||||
require_accelerate,
|
||||
require_hqq,
|
||||
require_torch_gpu,
|
||||
require_torch_multi_gpu,
|
||||
slow,
|
||||
@@ -86,6 +87,7 @@ MODEL_ID = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
||||
|
||||
|
||||
@require_torch_gpu
|
||||
@require_hqq
|
||||
class HqqConfigTest(unittest.TestCase):
|
||||
def test_to_dict(self):
|
||||
"""
|
||||
@@ -100,6 +102,7 @@ class HqqConfigTest(unittest.TestCase):
|
||||
@slow
|
||||
@require_torch_gpu
|
||||
@require_accelerate
|
||||
@require_hqq
|
||||
class HQQTest(unittest.TestCase):
|
||||
def tearDown(self):
|
||||
cleanup()
|
||||
@@ -122,6 +125,7 @@ class HQQTest(unittest.TestCase):
|
||||
@require_torch_gpu
|
||||
@require_torch_multi_gpu
|
||||
@require_accelerate
|
||||
@require_hqq
|
||||
class HQQTestMultiGPU(unittest.TestCase):
|
||||
def tearDown(self):
|
||||
cleanup()
|
||||
@@ -144,6 +148,7 @@ class HQQTestMultiGPU(unittest.TestCase):
|
||||
@slow
|
||||
@require_torch_gpu
|
||||
@require_accelerate
|
||||
@require_hqq
|
||||
class HQQSerializationTest(unittest.TestCase):
|
||||
def tearDown(self):
|
||||
cleanup()
|
||||
|
||||
Reference in New Issue
Block a user