Hugging Face's logo

Spaces:
optimum
/
neuron-exporter
Sleeping

App Files Community
2
badaoui HF Staff commited on
Commit
4c421ef
·
verified ·
1 Parent(s): 271a181

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -48,7 +48,10 @@ ARG OPTIMUM_NEURON_BRANCH=cpu-compilation
48
  ARG OPTIMUM_NEURON_REPO=https://github.com/huggingface/optimum-neuron.git
49
 
50
  # Install from your branch
51
- RUN pip install --no-cache-dir "git+https://github.com/huggingface/optimum-neuron.git@cpu-compilation[neuronx]"
 
 
 
52
 
53
  RUN pip install --no-cache-dir pip -U && \
54
  pip install --no-cache-dir \
 
48
  ARG OPTIMUM_NEURON_REPO=https://github.com/huggingface/optimum-neuron.git
49
 
50
  # Install from your branch
51
+ # install the branch
52
+ RUN pip install --no-cache-dir "git+https://github.com/huggingface/optimum-neuron.git@cpu-compilation#egg=optimum-neuron"
53
+ # add the extras
54
+ RUN pip install --no-cache-dir "optimum-neuron[neuronx]"
55
 
56
  RUN pip install --no-cache-dir pip -U && \
57
  pip install --no-cache-dir \