Update requirements.txt for finetune

This commit is contained in:
DejianYang 2024-04-16 10:16:07 +08:00 committed by GitHub
parent 7df235b013
commit b7ba565956
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,10 +1,10 @@
torch>=2.0.1
tokenizers>=0.14.0
torch==2.0.1
tokenizers==0.14.0
transformers==4.35.0
accelerate
accelerate==0.24.1
attrdict
tqdm
deepspeed
deepspeed==0.12.2
datasets
tensorboardX