diff --git a/.github/workflows/vllm_benchmark.yaml b/.github/workflows/vllm_benchmark.yaml index 03cae8ddeb..02818f996a 100644 --- a/.github/workflows/vllm_benchmark.yaml +++ b/.github/workflows/vllm_benchmark.yaml @@ -60,8 +60,9 @@ jobs: RUN echo "=== Aiter version BEFORE uninstall ===" && pip show aiter || true RUN pip uninstall -y aiter - RUN pip config set global.default-timeout 60 - RUN pip config set global.retries 10 + RUN pip config set global.default-timeout 60 \ + && pip config set global.retries 10 + RUN pip config set global.index-url https://ausartifactory.amd.com/artifactory/api/pypi/hw-cpe-prod-remote/simple RUN pip install --upgrade "pybind11>=3.0.1" RUN pip show pybind11 @@ -80,7 +81,7 @@ jobs: - name: Build Docker image run: | IMAGE_TAG=rocm/aiter-ci:${{ env.GITHUB_COMMIT_SHA }} - docker build -t $IMAGE_TAG -f Dockerfile.mod . + docker build --network=host --no-cache -t $IMAGE_TAG -f Dockerfile.mod . - name: Push Docker image run: |