tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fsdp2 support

Browse Source
This commit is contained in:
Wing Lian
2025-03-22 21:47:54 -04:00
parent a8f38c367c
commit 03dcf1a5ea
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@@ -14,7 +14,7 @@ packaging==23.2
peft==0.15.0
transformers==4.50.3
tokenizers>=0.21.1
accelerate==1.5.2
accelerate @ git+https://github.com/S1ro1/accelerate.git@dev/fsdp2
datasets==3.5.0
deepspeed==0.15.4
trl==0.16.0