This commit is contained in:
komaksym 2025-03-06 17:06:02 +01:00 committed by GitHub
commit d4bbc90119
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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