Bump deps

This commit is contained in:
Lewis Tunstall
2023-11-10 08:41:17 +00:00
parent 610a1a2de4
commit 0af8011993
+2 -2
View File
@@ -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",