diff --git a/requirements.txt b/requirements.txt index 0dbe372cf..3f8024dd5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,7 +8,7 @@ wandb==0.14.0 deepspeed>=0.14.4 trl==0.8.0 sentencepiece -transformers>=4.31.0 +transformers>=4.31.0, <=4.46.0 flask flask_cors icetk