skip quant test using bnb if bitsandbytes is not available (#649)

Lightning-AI · Jun 25, 2024 · 1aaa463 · 1aaa463
1 parent 0a0c7f4
commit 1aaa463
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/thunder/tests/test_networks.py b/thunder/tests/test_networks.py
@@ -208,6 +208,11 @@ def test_hf_bart_self_attn():
 
 @requiresCUDA
 def test_quantization():
+    try:
+        import bitsandbytes
+    except (ImportError, RuntimeError):
+        pytest.skip("bitsandbytes not found")
+
     from thunder.tests import litgpt_model
     from lightning.fabric.plugins import BitsandbytesPrecision