Spaces:

optimum
/

neuron-exporter

Running

badaoui HF Staff commited on Sep 17

Commit

3e25255

verified ·

1 Parent(s): 867ee4a

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

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

 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]"
 # End Neuron setup
 RUN pip install --no-cache-dir pip -U && \