Update BnB version

Lightning-AI · Dec 28, 2024 · 2c9260d · 2c9260d
1 parent 501846e
commit 2c9260d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -38,7 +38,8 @@ test = [
     "protobuf>=4.23.4",
 ]
 all = [
-    "bitsandbytes==0.42.0",      # quantization
+    "bitsandbytes >=0.44.0,<0.44.2; sys_platform == 'linux' or sys_platform == 'win32'", # quantization
+    "bitsandbytes >=0.42.0,<0.43.0 ; sys_platform == 'darwin'", # quantization
     "sentencepiece>=0.2.0",      # llama-based models
     "requests>=2.31.0",          # litgpt.data
     "litdata==0.2.17",           # litgpt.data