diff --git a/tests/quantization/quanto/test_quanto.py b/tests/quantization/quanto/test_quanto.py index 28555a6076b8..e3463f136f94 100644 --- a/tests/quantization/quanto/test_quanto.py +++ b/tests/quantization/quanto/test_quanto.py @@ -13,7 +13,7 @@ nightly, numpy_cosine_similarity_distance, require_accelerate, - require_big_accelerator, + require_accelerator, require_torch_cuda_compatibility, torch_device, ) @@ -31,7 +31,7 @@ @nightly -@require_big_accelerator +@require_accelerator @require_accelerate class QuantoBaseTesterMixin: model_id = None