diff --git a/.devops/main-cuda.Dockerfile b/.devops/main-cuda.Dockerfile index 7cd5ce0cf..b2650b9c5 100644 --- a/.devops/main-cuda.Dockerfile +++ b/.devops/main-cuda.Dockerfile @@ -38,7 +38,7 @@ RUN wget $MODEL_URL -O /model.gguf WORKDIR /install RUN apt-get install -y python3 python3-pip -RUN pip install --install-option="--prefix=/install" runpod +RUN pip install --prefix /install runpod FROM ${BASE_CUDA_RUN_CONTAINER} as runtime