Merge pull request #105 from huggingface/dev/neuronx-versions

fixing neuronx dependencies + optimum upgrade

Author: oOraph

dockerfiles/pytorch/Dockerfile.inf2

@@ -8,16 +8,17 @@ ARG PYTHON_VERSION=3.10.12
 ARG MAMBA_VERSION=23.1.0-4
 
 # Neuron SDK components version numbers
-ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.1.0
-ARG NEURONX_DISTRIBUTED_VERSION=0.7.0
-ARG NEURONX_CC_VERSION=2.13.66.0
-ARG NEURONX_TRANSFORMERS_VERSION=0.10.0.21
-ARG NEURONX_COLLECTIVES_LIB_VERSION=2.20.22.0-c101c322e
-ARG NEURONX_RUNTIME_LIB_VERSION=2.20.22.0-1b3ca6425
-ARG NEURONX_TOOLS_VERSION=2.17.1.0
+# ARG NEURONX_FRAMEWORK_VERSION=2.1.2.2.1.0
+# ARG NEURONX_DISTRIBUTED_VERSION=0.7.0
+# ARG NEURONX_CC_VERSION=2.13.66.0
+ARG NEURONX_TRANSFORMERS_VERSION=0.12.313
+ARG NEURONX_COLLECTIVES_LIB_VERSION=2.22.33.0-d2128d1aa
+ARG NEURONX_RUNTIME_LIB_VERSION=2.22.19.0-5856c0b42
+ARG NEURONX_TOOLS_VERSION=2.19.0.0
+
 
 # HF ARGS
-ARG OPTIMUM_NEURON_VERSION=0.0.23
+ARG OPTIMUM_NEURON_VERSION=0.0.28
 
 # See http://bugs.python.org/issue19846
 ENV LANG C.UTF-8