Fix pinned huggingface-inference-toolkit to include path patch (#127)

huggingface · Dec 3, 2024 · b174d63 · b174d63
1 parent edc876d
commit b174d63
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/containers/pytorch/inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile b/containers/pytorch/inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile
@@ -34,8 +34,8 @@ RUN apt-get update && \
 # Prevent `huggingface_hub>0.26.0` from being installed later on
 RUN pip install "huggingface_hub[hf_transfer]<0.26.0"
 
-# Hugging Face Inference Toolkit
-ARG HF_INFERENCE_TOOLKIT_VERSION=0.4.1
+# Hugging Face Inference Toolkit (version 0.4.1 + path fix)
+ARG HF_INFERENCE_TOOLKIT_VERSION=58b760fe3ec4cbddf064ac62c7a3f745af136d5f
 ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
 RUN pip install "${HF_INFERENCE_TOOLKIT_URL}#egg=huggingface-inference-toolkit[torch,diffusers,st]"
 

diff --git a/containers/pytorch/inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile b/containers/pytorch/inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile
@@ -34,8 +34,8 @@ RUN apt-get update && \
 # Prevent `huggingface_hub>0.26.0` from being installed later on
 RUN pip install "huggingface_hub[hf_transfer]<0.26.0"
 
-# Hugging Face Inference Toolkit
-ARG HF_INFERENCE_TOOLKIT_VERSION=0.4.1
+# Hugging Face Inference Toolkit (version 0.4.1 + path fix)
+ARG HF_INFERENCE_TOOLKIT_VERSION=58b760fe3ec4cbddf064ac62c7a3f745af136d5f
 ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
 RUN pip install "${HF_INFERENCE_TOOLKIT_URL}#egg=huggingface-inference-toolkit[torch,diffusers,st]"