diff --git a/model/supervised_finetuning/requirements.txt b/model/supervised_finetuning/requirements.txt index 6338614d..c47a1218 100644 --- a/model/supervised_finetuning/requirements.txt +++ b/model/supervised_finetuning/requirements.txt @@ -1,4 +1,5 @@ accelerate==0.15.0 +bitsandbytes==0.36.0.post2 datasets==2.8.0 deepspeed==0.7.7 mpi4py==3.1.4 @@ -7,4 +8,3 @@ PyYAML==6.0 scikit_learn==1.2.0 torch==1.13.1 transformers==4.25.1 -bitsandbytes==0.36.0.post2 \ No newline at end of file