diff --git a/docker/Dockerfile b/docker/Dockerfile index b264d58..8d6cea6 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -309,7 +309,9 @@ RUN set -ex && \ ${VENV}/bin/pip install --force-reinstall \ torch==2.11.0+rocm7.2 torchvision pytorch-triton-rocm \ --index-url ${TORCH_INDEX} && \ - cd /opt/vllm && ${VENV}/bin/pip install -e . --no-build-isolation --no-deps && \ + cd /opt/vllm && \ + sed -i 's/license = "Apache-2.0"/license = {text = "Apache-2.0"}/' pyproject.toml && \ + ${VENV}/bin/pip install -e . --no-build-isolation --no-deps && \ ${VENV}/bin/python -c "import triton.language; print('triton OK')" && \ ${VENV}/bin/python -c "import torch; v=torch.__version__; assert 'rocm' in v, f'PyTorch overwritten: {v}'; print('PyTorch OK:', v)" && \ ${VENV}/bin/python -c "import vllm; print('vllm OK:', vllm.__version__)"