Skip to content

Commit bcd4d77

Browse files
authored
[CI] Remove big accelerator requirements from Quanto Tests (huggingface#12266)
update
1 parent 006d092 commit bcd4d77

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/quantization/quanto/test_quanto.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
nightly,
1414
numpy_cosine_similarity_distance,
1515
require_accelerate,
16-
require_big_accelerator,
16+
require_accelerator,
1717
require_torch_cuda_compatibility,
1818
torch_device,
1919
)
@@ -31,7 +31,7 @@
3131

3232

3333
@nightly
34-
@require_big_accelerator
34+
@require_accelerator
3535
@require_accelerate
3636
class QuantoBaseTesterMixin:
3737
model_id = None

0 commit comments

Comments
 (0)