upgrade torch neuronx following the 2.16.0 guideline (#1517)

deepjavalibrary · Jan 27, 2024 · b909f12 · b909f12
1 parent a959555
commit b909f12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile
@@ -14,7 +14,7 @@ ARG djl_version=0.26.0
 ARG torch_version=1.13.1
 ARG python_version=3.9
 ARG neuronsdk_version=2.16.0
-ARG torch_neuronx_version=1.13.1.1.12.1
+ARG torch_neuronx_version=1.13.1.1.13.0
 ARG transformers_neuronx_version=0.9.474
 ARG neuronx_distributed_version=0.6.0
 ARG neuronx_cc_version=2.12.54.0