mirror of
https://github.com/deepseek-ai/DeepSeek-Coder
synced 2025-04-05 21:15:14 +00:00
fixed finetune requirements.txt
This commit is contained in:
parent
b7ba565956
commit
ff8432500b
4
finetune/install.sh
Executable file
4
finetune/install.sh
Executable file
@ -0,0 +1,4 @@
|
||||
#!/bin/bash
|
||||
pip install -r requirements.txt
|
||||
DS_BUILD_CPU_ADAM=1 pip install deepspeed==0.14.5
|
||||
|
@ -2,9 +2,9 @@ torch==2.0.1
|
||||
tokenizers==0.14.0
|
||||
transformers==4.35.0
|
||||
accelerate==0.24.1
|
||||
numpy==1.26.4
|
||||
attrdict
|
||||
tqdm
|
||||
|
||||
deepspeed==0.12.2
|
||||
datasets
|
||||
tensorboardX
|
||||
|
Loading…
Reference in New Issue
Block a user