Spaces:

optimum
/

neuron-exporter

Running

badaoui HF Staff commited on Oct 3

Commit

ec02103

verified ·

1 Parent(s): 5964425

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -45,7 +45,8 @@ ENV PATH =/opt/aws/neuron/bin:$PATH
 ARG OPTIMUM_NEURON_BRANCH=main
 RUN git clone --branch $OPTIMUM_NEURON_BRANCH https://github.com/huggingface/optimum-neuron.git /tmp/optimum-neuron && \
-    pip install --no-cache-dir ".[neuronx]" /tmp/optimum-neuron
 # End Neuron setup
 RUN pip install --no-cache-dir pip -U && \

 ARG OPTIMUM_NEURON_BRANCH=main
 RUN git clone --branch $OPTIMUM_NEURON_BRANCH https://github.com/huggingface/optimum-neuron.git /tmp/optimum-neuron && \
+    pip install --no-cache-dir "/tmp/optimum-neuron[neuronx]" && \
+    rm -rf /tmp/optimum-neuron
 # End Neuron setup
 RUN pip install --no-cache-dir pip -U && \