update

2025-12-25 05:44:52 +08:00 · 2025-09-02 09:18:23 +05:30
1 changed files with 2 additions and 2 deletions
--- 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