mirror of
https://github.com/deepseek-ai/DeepSeek-Coder
synced 2025-04-22 07:05:27 +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
|
tokenizers==0.14.0
|
||||||
transformers==4.35.0
|
transformers==4.35.0
|
||||||
accelerate==0.24.1
|
accelerate==0.24.1
|
||||||
|
numpy==1.26.4
|
||||||
attrdict
|
attrdict
|
||||||
tqdm
|
tqdm
|
||||||
|
|
||||||
deepspeed==0.12.2
|
|
||||||
datasets
|
datasets
|
||||||
tensorboardX
|
tensorboardX
|
||||||
|
Loading…
Reference in New Issue
Block a user