🐛 fixup release stage

Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
IBM · Apr 11, 2024 · 805e003 · 805e003
1 parent 481b1c6
commit 805e003
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Dockerfile.ubi b/Dockerfile.ubi
@@ -281,9 +281,10 @@ COPY --from=python-torch-base --link /opt/vllm /opt/vllm
 ENV PATH=/opt/vllm/bin/:$PATH
 
 RUN --mount=type=cache,target=/root/.cache/pip \
-    --mount=type=bind,source=requirements.txt,target=requirements.txt \
+    --mount=type=bind,source=requirements-common.txt,target=requirements-common.txt \
+    --mount=type=bind,source=requirements-cuda.txt,target=requirements-cuda.txt \
     pip3 install \
-        -r requirements.txt \
+        -r requirements-cuda.txt \
         # additional dependencies for the TGIS gRPC server
         grpcio-tools==1.62.1 \
         # additional dependencies for openai api_server