Pin bnb to <=0.41.1 (#968)

* pin bnb to 0.41.1 * Update setup.py * Update setup.py
huggingface · Nov 8, 2023 · 1b25824 · 1b25824
1 parent 9c93dec
commit 1b25824
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -73,7 +73,7 @@
     "diffusers": ["diffusers>=0.18.0"],
     "deepspeed": ["deepspeed>=0.9.5"],
     "benchmark": ["wandb", "ghapi", "openrlbenchmark==0.2.1a5", "requests", "deepspeed"],
-    "quantization": ["bitsandbytes>=0.41.0"],
+    "quantization": ["bitsandbytes<=0.41.1"],
 }
 EXTRAS["dev"] = []
 for reqs in EXTRAS.values():