diff --git a/fine_tune_requirements.txt b/fine_tune_requirements.txt index 9a03c5d..0cb0020 100644 --- a/fine_tune_requirements.txt +++ b/fine_tune_requirements.txt @@ -1,7 +1,7 @@ peft==0.5.0 transformers==4.37.1 transformers-stream-generator==0.0.4 -deepspeed==0.9.5 +deepspeed==0.14.1 accelerate==0.26.1 gunicorn==20.1.0 flask==2.1.2