From bcd4d77ba61c9ec295d0649448f108866b766708 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Wed, 3 Sep 2025 04:59:31 +0200 Subject: [PATCH] [CI] Remove big accelerator requirements from Quanto Tests (#12266) update --- tests/quantization/quanto/test_quanto.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/quantization/quanto/test_quanto.py b/tests/quantization/quanto/test_quanto.py index 28555a6076..e3463f136f 100644 --- a/tests/quantization/quanto/test_quanto.py +++ b/tests/quantization/quanto/test_quanto.py @@ -13,7 +13,7 @@ from ...testing_utils import ( nightly, numpy_cosine_similarity_distance, require_accelerate, - require_big_accelerator, + require_accelerator, require_torch_cuda_compatibility, torch_device, ) @@ -31,7 +31,7 @@ enable_full_determinism() @nightly -@require_big_accelerator +@require_accelerator @require_accelerate class QuantoBaseTesterMixin: model_id = None