fixed finetune requirements.txt

This commit is contained in:
komaksym 2025-03-06 05:27:21 +01:00
parent b7ba565956
commit ff8432500b
2 changed files with 5 additions and 1 deletions

4
finetune/install.sh Executable file
View File

@ -0,0 +1,4 @@
#!/bin/bash
pip install -r requirements.txt
DS_BUILD_CPU_ADAM=1 pip install deepspeed==0.14.5

View File

@ -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