DeepSeek/DeepSeek-Coder
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Coder synced 2025-06-26 18:25:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update requirements.txt for finetune

Browse Source
This commit is contained in:
DejianYang
2024-04-16 10:16:07 +08:00
committed by GitHub
parent 7df235b013
commit b7ba565956
1 changed files with 4 additions and 4 deletions
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