diff --git a/tests/quantization/bnb/test_mixed_int8.py b/tests/quantization/bnb/test_mixed_int8.py index 981881c005b7..a67d26d8cad4 100644 --- a/tests/quantization/bnb/test_mixed_int8.py +++ b/tests/quantization/bnb/test_mixed_int8.py @@ -41,6 +41,7 @@ load_pt, numpy_cosine_similarity_distance, require_accelerate, + require_big_accelerator, require_bitsandbytes_version_greater, require_peft_backend, require_peft_version_greater, @@ -590,6 +591,7 @@ def get_dummy_tensor_inputs(device=None, seed: int = 0): @require_transformers_version_greater("4.44.0") +@require_big_accelerator class TestSlowBnb8bitFlux(Base8bitTests): @pytest.fixture(autouse=True) def _setup_slow_flux(self):