Skip to content

Commit e0cee9b

Browse files
committed
Update pip install syntax when installing from URL
1 parent 08f4738 commit e0cee9b

File tree

2 files changed

+2
-2
lines changed
  • containers/pytorch/inference

2 files changed

+2
-2
lines changed

containers/pytorch/inference/cpu/2.3.1/transformers/4.46.0/py311/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
4242
# Hugging Face Inference Toolkit
4343
ARG HF_INFERENCE_TOOLKIT_VERSION=bump-dependencies
4444
ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
45-
RUN pip install --upgrade "${HF_INFERENCE_TOOLKIT_URL}#egg=huggingface-inference-toolkit[torch,diffusers,st,google]" --no-cache-dir
45+
RUN pip install --upgrade "huggingface-inference-toolkit[torch,diffusers,st,google] @ ${HF_INFERENCE_TOOLKIT_URL}" --no-cache-dir
4646

4747
ENV HF_HUB_ENABLE_HF_TRANSFER="1"
4848

containers/pytorch/inference/gpu/2.3.1/transformers/4.46.0/py311/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
4242
# Hugging Face Inference Toolkit
4343
ARG HF_INFERENCE_TOOLKIT_VERSION=bump-dependencies
4444
ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
45-
RUN pip install --upgrade "huggingface-inference-toolkit[torch,diffusers,st,google]" @ ${HF_INFERENCE_TOOLKIT_URL} --no-cache-dir
45+
RUN pip install --upgrade "huggingface-inference-toolkit[torch,diffusers,st,google] @ ${HF_INFERENCE_TOOLKIT_URL}" --no-cache-dir
4646

4747
ENV HF_HUB_ENABLE_HF_TRANSFER="1"
4848

0 commit comments

Comments
 (0)