Update

deepjavalibrary · May 14, 2024 · c15e5ad · c15e5ad
1 parent 8353588
commit c15e5ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/serving/docker/lmi.Dockerfile b/serving/docker/lmi.Dockerfile
@@ -94,8 +94,8 @@ RUN pip3 install torch==${torch_version} torchvision==${torch_vision_version} --
     mpi4py sentencepiece tiktoken blobfile einops accelerate==${accelerate_version} bitsandbytes==${bitsandbytes_version} \
     optimum==${optimum_version} auto-gptq==${auto_gptq_version} pandas pyarrow jinja2 \
     opencv-contrib-python-headless safetensors scipy \
-    onnxruntime-gpu==${onnx_version} --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ \
-    ${djl_converter_wheel} && \
+    onnxruntime-gpu==${onnx_version} --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ && \
+    pip3 install ${djl_converter_wheel} --no-deps && \
     pip3 cache purge
 
 RUN pip3 install ${flash_attn_2_wheel} ${lmi_dist_wheel} ${vllm_wheel} pydantic==${pydantic_version} && \