NW
/
DeepSeek-Coder
mirror of https://github.com/deepseek-ai/DeepSeek-Coder
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update requirements.txt for finetune

This commit is contained in:
DejianYang 2024-04-16 10:16:07 +08:00 committed by GitHub
parent 7df235b013
commit b7ba565956
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
finetune/requirements.txt
View File

@ -1,10 +1,10 @@
torch>=2.0.1
tokenizers>=0.14.0
torch==2.0.1
tokenizers==0.14.0
transformers==4.35.0
accelerate
accelerate==0.24.1
attrdict
tqdm
deepspeed
deepspeed==0.12.2
datasets
tensorboardX