diff --git a/setup.py b/setup.py index 1d2af6a..cdb4e1b 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ if stale_egg_info.exists(): # * If a dependency is fast-moving (e.g. transformers), pin to the exact version _deps = [ "accelerate==0.23.0", - "bitsandbytes==0.41.1", + "bitsandbytes==0.41.2.post2", "black==23.1.0", "datasets==2.14.6", "deepspeed==0.12.2", @@ -56,7 +56,7 @@ _deps = [ "numpy>=1.24.2", "packaging>=23.0", "parameterized>=0.9.0", - "peft==0.6.0", + "peft==0.6.1", "protobuf<=3.20.2", # Needed to avoid conflicts with `transformers` "pytest", "safetensors>=0.3.3",