Add fsdp+qlora support (#160)

This commit is contained in:
Chansung Park
2024-05-08 22:08:13 +09:00
committed by GitHub
parent 84f8c92820
commit 606d2e954f
5 changed files with 35 additions and 2 deletions
+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.29.2",
"bitsandbytes==0.41.2.post2",
"bitsandbytes>=0.43.0",
"black==23.1.0",
"datasets>=2.18.0",
"deepspeed==0.12.2",
@@ -57,7 +57,7 @@ _deps = [
"numpy>=1.24.2",
"packaging>=23.0",
"parameterized>=0.9.0",
"peft==0.7.1",
"peft>=0.9.0",
"protobuf<=3.20.2", # Needed to avoid conflicts with `transformers`
"pytest",
"safetensors>=0.3.3",