Skip to content

Commit

Permalink
[docker] bump neuron sdk packages to 2.20.2
Browse files Browse the repository at this point in the history
  • Loading branch information
sindhuvahinis committed Nov 25, 2024
1 parent 3d39409 commit 003d216
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions serving/docker/pytorch-inf2.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -15,13 +15,13 @@ ARG djl_serving_version
ARG torch_version=2.1.2
ARG torchvision_version=0.16.2
ARG python_version=3.10
ARG neuronsdk_version=2.20.1
ARG torch_neuronx_version=2.1.2.2.3.1
ARG neuronsdk_version=2.20.2
ARG torch_neuronx_version=2.1.2.2.3.2
ARG transformers_neuronx_version=0.12.313
ARG neuronx_distributed_version=0.9.0
ARG neuronx_cc_version=2.15.141.0
ARG neuronx_cc_stubs_version=2.15.141.0
ARG torch_xla_version=2.1.4
ARG neuronx_cc_version=2.15.143.0
ARG neuronx_cc_stubs_version=2.15.143.0
ARG torch_xla_version=2.1.5
ARG transformers_version=4.45.2
ARG accelerate_version=0.29.2
ARG diffusers_version=0.28.2
Expand Down

0 comments on commit 003d216

Please sign in to comment.