badaoui HF Staff commited on
Commit
529bfb9
·
verified ·
1 Parent(s): 4c421ef

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -49,7 +49,7 @@ 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
 
 
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"
53
  # add the extras
54
  RUN pip install --no-cache-dir "optimum-neuron[neuronx]"
55