diff --git a/src/requirements.txt b/src/requirements.txt index fd60b42..3672c9c 100644 --- a/src/requirements.txt +++ b/src/requirements.txt @@ -2,7 +2,7 @@ packaging xformers==0.0.21 tqdm sentencepiece -transformer==4.31.0 +transformers==4.31.0 datasets==2.14.5 deepspeed==0.11.0 accelerate==0.21.0