diff --git a/finetune/requirements.txt b/finetune/requirements.txt index 07a0685..bc1d43e 100644 --- a/finetune/requirements.txt +++ b/finetune/requirements.txt @@ -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