badaoui HF Staff commited on
Commit
271a181
·
verified ·
1 Parent(s): 9e6236d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -48,7 +48,7 @@ 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+$OPTIMUM_NEURON_REPO@$OPTIMUM_NEURON_BRANCH#egg=optimum-neuron[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
+ 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 \