Update Dockerfile

Dockerfile

@@ -43,9 +43,10 @@ RUN apt-get update && apt-get install -y \
 
 ENV PATH =/opt/aws/neuron/bin:$PATH
 
-RUN pip install --extra-index-url https://pip.repos.neuron.amazonaws.com --no-cache-dir \
-    "git+https://github.com/huggingface/optimum-neuron.git@cpu-compilation#egg=optimum-neuron[neuronx]"
-
+RUN pip install --no-cache-dir --extra-index-url https://pip.repos.neuron.amazonaws.com \
+    git+https://github.com/huggingface/optimum-neuron.git@cpu-compilation
+RUN pip install --no-cache-dir --extra-index-url https://pip.repos.neuron.amazonaws.com \
+    optimum-neuron[neuronx]
 
 # End Neuron setup
 RUN pip install --no-cache-dir pip -U && \