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 torch==2.0.1
tokenizers>=0.14.0 tokenizers==0.14.0
transformers==4.35.0 transformers==4.35.0
accelerate accelerate==0.24.1
attrdict attrdict
tqdm tqdm
deepspeed deepspeed==0.12.2
datasets datasets
tensorboardX tensorboardX