From d95c741ea0b109f45471af823782af35a51cf04f Mon Sep 17 00:00:00 2001 From: mrcabbage972 Date: Tue, 10 Jan 2023 20:16:02 -0500 Subject: [PATCH] Fixing requirements file --- model/supervised_finetuning/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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