Update pip install syntax when installing from URL

huggingface · Oct 25, 2024 · e0cee9b · e0cee9b
1 parent 08f4738
commit e0cee9b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/containers/pytorch/inference/cpu/2.3.1/transformers/4.46.0/py311/Dockerfile b/containers/pytorch/inference/cpu/2.3.1/transformers/4.46.0/py311/Dockerfile
@@ -42,7 +42,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
 # Hugging Face Inference Toolkit
 ARG HF_INFERENCE_TOOLKIT_VERSION=bump-dependencies
 ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
-RUN pip install --upgrade "${HF_INFERENCE_TOOLKIT_URL}#egg=huggingface-inference-toolkit[torch,diffusers,st,google]" --no-cache-dir
+RUN pip install --upgrade "huggingface-inference-toolkit[torch,diffusers,st,google] @ ${HF_INFERENCE_TOOLKIT_URL}" --no-cache-dir
 
 ENV HF_HUB_ENABLE_HF_TRANSFER="1"
 

diff --git a/containers/pytorch/inference/gpu/2.3.1/transformers/4.46.0/py311/Dockerfile b/containers/pytorch/inference/gpu/2.3.1/transformers/4.46.0/py311/Dockerfile
@@ -42,7 +42,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
 # Hugging Face Inference Toolkit
 ARG HF_INFERENCE_TOOLKIT_VERSION=bump-dependencies
 ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
-RUN pip install --upgrade "huggingface-inference-toolkit[torch,diffusers,st,google]" @ ${HF_INFERENCE_TOOLKIT_URL} --no-cache-dir
+RUN pip install --upgrade "huggingface-inference-toolkit[torch,diffusers,st,google] @ ${HF_INFERENCE_TOOLKIT_URL}" --no-cache-dir
 
 ENV HF_HUB_ENABLE_HF_TRANSFER="1"