Skip to content

Commit 293da5a

Browse files
[docker] update vllm wheel in neuronx container (#2688)
1 parent bf4d851 commit 293da5a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

serving/docker/pytorch-inf2.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ ARG pydantic_version=2.6.1
3737
ARG optimum_neuron_version=0.0.27
3838
ARG huggingface_hub_version=0.25.2
3939
# %2B is the url escape for the '+' character
40-
ARG vllm_wheel="https://publish.djl.ai/neuron_vllm/vllm-0.6.2%2Bnightly-py3-none-any.whl"
40+
ARG vllm_wheel="https://publish.djl.ai/neuron_vllm/vllm_neuronx-0.6.2%2Bneuron216-py3-none-any.whl"
4141
EXPOSE 8080
4242

4343
# Sets up Path for Neuron tools

0 commit comments

Comments
 (0)