Set huggingface-inference-toolkit released version

huggingface · Jan 18, 2025 · 56fa8a8 · 56fa8a8
1 parent e4ca220
commit 56fa8a8
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.48.0/py311/Dockerfile b/containers/pytorch/inference/cpu/2.3.1/transformers/4.48.0/py311/Dockerfile
@@ -40,7 +40,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
     rm get-pip.py
 
 # Hugging Face Inference Toolkit
-ARG HF_INFERENCE_TOOLKIT_VERSION=bump-dependencies
+ARG HF_INFERENCE_TOOLKIT_VERSION=0.5.4
 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
 

diff --git a/containers/pytorch/inference/gpu/2.3.1/transformers/4.48.0/py311/Dockerfile b/containers/pytorch/inference/gpu/2.3.1/transformers/4.48.0/py311/Dockerfile
@@ -40,7 +40,7 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
     rm get-pip.py
 
 # Hugging Face Inference Toolkit
-ARG HF_INFERENCE_TOOLKIT_VERSION=bump-dependencies
+ARG HF_INFERENCE_TOOLKIT_VERSION=0.5.4
 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