diff --git a/Dockerfile_amd b/Dockerfile_amd index e3e9efda8a2..f2ef913e7c0 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -251,7 +251,7 @@ COPY proto proto COPY server server COPY server/Makefile server/Makefile RUN cd server && \ - uv pip install grpcio-tools mypy-protobuf && \ + uv pip install "grpcio-tools>=1.67.0,<1.73.0" "grpcio>=1.67.0,<1.73.0" mypy-protobuf && \ uv pip install -e ".[accelerate, compressed-tensors, peft, outlines]" --no-cache-dir && \ make gen-server-raw RUN cd server && \ diff --git a/server/pyproject.toml b/server/pyproject.toml index 7f2addb69a1..74615b85b0e 100644 --- a/server/pyproject.toml +++ b/server/pyproject.toml @@ -13,9 +13,9 @@ dependencies = [ "click<8.2.0", "einops>=0.8.0", "grpc-interceptor>=0.15.4", - "grpcio>=1.67.0", - "grpcio-reflection>=1.67.0", - "grpcio-status>=1.67.0", + "grpcio>=1.67.0,<1.73.0", + "grpcio-reflection>=1.67.0,<1.73.0", + "grpcio-status>=1.67.0,<1.73.0", "kernels>=0.2.1", "hf-transfer>=0.1.8", "loguru>=0.7.3", @@ -80,7 +80,7 @@ peft = [ "peft>=0.14.0", ] outlines = [ - "outlines>=0.1.13", + "outlines>=0.1.13,<1.0", ] dev = [ "grpcio-tools>=1.51.1,<2.0",