Spaces:
Running
Running
Update Dockerfile
Browse files- 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+
|
| 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 \
|