diff --git a/docker/inference/requirements.txt b/docker/inference/requirements.txt index bb03659..29058d8 100644 --- a/docker/inference/requirements.txt +++ b/docker/inference/requirements.txt @@ -3,7 +3,7 @@ fastapi==0.111.0 uvicorn==0.30.1 gunicorn==22.0.0 -transformers==4.53.0 +transformers==5.0.0rc3 # transformers 4.53.0 requires huggingface-hub >=0.30.0,<1.0 huggingface_hub==0.30.0 torch==2.8.0 diff --git a/docker/training/requirements.txt b/docker/training/requirements.txt index 22b6492..1ec1c6d 100644 --- a/docker/training/requirements.txt +++ b/docker/training/requirements.txt @@ -1,5 +1,5 @@ # Training dependencies (extend inference set if needed) -transformers==4.53.0 +transformers==5.0.0rc3 # transformers 4.53.0 requires huggingface-hub >=0.30.0,<1.0 huggingface_hub==0.30.0 torch==2.8.0