diff --git a/docker/Dockerfile.ci b/docker/Dockerfile.ci index a7c3beb9d..bd7ccd5fd 100644 --- a/docker/Dockerfile.ci +++ b/docker/Dockerfile.ci @@ -34,17 +34,6 @@ RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm # Install vllm-omni into the same uv-managed Python environment used by the base image. RUN uv pip install --python "$(python3 -c 'import sys; print(sys.executable)')" --no-cache-dir ".[dev]" -# Debug: print key dependency versions for CI troubleshooting -RUN python3 -c "\ -import transformers, torch, vllm; \ -print(f'transformers={transformers.__version__}'); \ -print(f'torch={torch.__version__}'); \ -print(f'vllm={vllm.__version__}'); \ -print(f'cuda={torch.version.cuda}'); \ -" - -RUN uv pip install transformers==4.57.6 - RUN ln -sf /usr/bin/python3 /usr/bin/python ENTRYPOINT [] diff --git a/pyproject.toml b/pyproject.toml index 6af193953..34a7c977e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -48,7 +48,8 @@ dev = [ "soundfile>=0.13.1", "imageio[ffmpeg]>=0.6.0", "opencv-python>=4.12.0.88", - "mooncake-transfer-engine==0.3.8.post1" + "mooncake-transfer-engine==0.3.8.post1", + "transformers==4.57.6" ] docs = [