Update setup.py

huggingface · Nov 8, 2023 · 351f865 · 351f865
1 parent a40ca55
commit 351f865
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.1"],
+    "quantization": ["bitsandbytes<=0.41.1"],
 }
 EXTRAS["dev"] = []
 for reqs in EXTRAS.values():