set max torch for bnb

Lightning-AI · Jun 30, 2024 · 65b86c5 · 65b86c5
1 parent 7524247
commit 65b86c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/tests_fabric/plugins/precision/test_bitsandbytes.py b/tests/tests_fabric/plugins/precision/test_bitsandbytes.py
@@ -93,7 +93,7 @@ def __init__(self):
         precision.convert_module(model)
 
 
-@RunIf(min_cuda_gpus=1)
+@RunIf(min_cuda_gpus=1, max_torch="2.4")
 @pytest.mark.skipif(not _BITSANDBYTES_AVAILABLE, reason="bitsandbytes unavailable")
 @pytest.mark.parametrize(
     ("args", "expected"),
@@ -232,7 +232,7 @@ def __init__(self):
     assert model.l.weight.dtype == expected
 
 
-@RunIf(min_cuda_gpus=1, min_torch="2.1")
+@RunIf(min_cuda_gpus=1, min_torch="2.1", max_torch="2.4")
 @pytest.mark.skipif(not _BITSANDBYTES_AVAILABLE, reason="bitsandbytes unavailable")
 def test_load_quantized_checkpoint(tmp_path):
     """Test that a checkpoint saved from a quantized model can be loaded back into a quantized model."""