diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 249ca72981..a1b0d6b5b3 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -25,16 +25,20 @@ pydantic>=2.10 fastapi matplotlib librosa -torch==2.12 +torch==2.13 cuda-bindings>=13.1.0 # Required for CUDA graph annotations tutorial torchvision torchdata networkx PyHamcrest bs4 +# Pin setuptools: awscliv2==2.1.1 imports pkg_resources, which setuptools>=81 removed. +setuptools<81 awscliv2==2.1.1 flask spacy==3.7.1 # Keep this version consistent with the model version in .jenkins/build.sh +# Pin protobuf: ray[serve]==2.55.0 uses FieldDescriptor.label, removed in protobuf>=6. +protobuf<6 ray[serve,train,tune]==2.55.0 tiktoken tensorboard