diff --git a/online-inference/bloom-176b-deepspeed/files/requirements.txt b/online-inference/bloom-176b-deepspeed/files/requirements.txt index 10c6f909..f6fd3b0e 100644 --- a/online-inference/bloom-176b-deepspeed/files/requirements.txt +++ b/online-inference/bloom-176b-deepspeed/files/requirements.txt @@ -2,7 +2,7 @@ torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 -transformers==4.21.3 +transformers==4.48.0 accelerate==0.12.0 deepspeed>=0.7.3 flask==2.2.2