♻️ gen protos befiore buikding wheel

Signed-off-by: Prashant Gupta <[email protected]>

Dockerfile.ubi

@@ -192,6 +192,8 @@ RUN python3 setup.py build_ext --inplace
 
 # Copy the entire directory before building wheel
 COPY vllm vllm
+# Copy over the generated *.pb2 files
+COPY --from=gen-protos --link /workspace/vllm/entrypoints/grpc/pb vllm/entrypoints/grpc/pb
 
 RUN --mount=type=cache,target=/root/.cache/ccache \
     --mount=type=cache,target=/root/.cache/pip \
@@ -204,7 +206,6 @@ RUN --mount=type=cache,target=/root/.cache/ccache \
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip cache remove vllm_nccl*
 
-
 ## Extension Cache #############################################################
 # Instead of compiling artifacts every build just copy from pre-built wheel
 # This might not work if the PyTorch and CUDA versions don't match!
@@ -286,8 +287,6 @@ WORKDIR /workspace
 COPY --from=vllm --link /opt/vllm /opt/vllm
 ENV PATH=/opt/vllm/bin/:$PATH
 
-COPY --from=gen-protos --link /workspace/vllm/entrypoints/grpc/pb vllm/entrypoints/grpc/pb
-
 RUN --mount=type=cache,target=/root/.cache/pip \
     --mount=type=bind,source=requirements-common.txt,target=requirements-common.txt \
     --mount=type=bind,source=requirements-cuda.txt,target=requirements-cuda.txt \