From bf4b24bfd4d3c6d5ee760c0dc72cd7a8a7784ecb Mon Sep 17 00:00:00 2001 From: Somasundaram Date: Mon, 4 Nov 2024 11:04:31 -0800 Subject: [PATCH] [docker] upgrade neuron vllm to 0.6.2 --- serving/docker/pytorch-inf2.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile index 8665b9f5eb..1cc5f5689d 100644 --- a/serving/docker/pytorch-inf2.Dockerfile +++ b/serving/docker/pytorch-inf2.Dockerfile @@ -29,7 +29,7 @@ ARG pydantic_version=2.6.1 ARG optimum_neuron_version=0.0.24 ARG huggingface_hub_version=0.25.2 # %2B is the url escape for the '+' character -ARG vllm_wheel="https://publish.djl.ai/neuron_vllm/vllm-0.6.0%2Bnightly-py3-none-any.whl" +ARG vllm_wheel="https://publish.djl.ai/neuron_vllm/vllm-0.6.2%2Bnightly-py3-none-any.whl" EXPOSE 8080 # Sets up Path for Neuron tools