From b174d631b9106d1e869f634c1839aff2cd0d7bac Mon Sep 17 00:00:00 2001
From: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com>
Date: Tue, 3 Dec 2024 09:48:53 +0100
Subject: [PATCH] Fix pinned `huggingface-inference-toolkit` to include path
 patch (#127)

---
 .../inference/cpu/2.2.2/transformers/4.41.1/py311/Dockerfile  | 4 ++--
 .../inference/gpu/2.2.2/transformers/4.41.1/py311/Dockerfile  | 4 ++--
 2 files changed, 4 insertions(+), 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
index fdaff531..73456c6c 100644
--- 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
index 69047fc5..848afef3 100644
--- 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]"