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
index 94bd4884..eb4a557f 100644
--- 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
index 0baf400f..066d71a2 100644
--- 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"