Skip to content

Commit 8d17324

Browse files
committed
Remove uv and don't upgrade setuptools
Just by fixing the Python 3.11 and the `pip` installation, the installation issue affecting `kenlm` is solved already; so no need to add `uv` for the moment even though it would be a nice addition
1 parent b6e102e commit 8d17324

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

containers/pytorch/inference/gpu/2.3.1/transformers/4.46.0/py311/Dockerfile

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,14 +39,10 @@ RUN curl -O https://bootstrap.pypa.io/get-pip.py && \
3939
python get-pip.py && \
4040
rm get-pip.py
4141

42-
# Install uv and upgrade setuptools
43-
RUN pip install --upgrade uv && \
44-
uv pip install --upgrade setuptools --no-cache-dir --system
45-
4642
# Hugging Face Inference Toolkit
4743
ARG HF_INFERENCE_TOOLKIT_VERSION=bump-dependencies
4844
ARG HF_INFERENCE_TOOLKIT_URL=git+https://github.com/huggingface/huggingface-inference-toolkit.git@${HF_INFERENCE_TOOLKIT_VERSION}
49-
RUN uv pip install --upgrade "${HF_INFERENCE_TOOLKIT_URL}#egg=huggingface-inference-toolkit[torch,diffusers,st,google]" --no-cache-dir --system
45+
RUN pip install --upgrade "huggingface-inference-toolkit[torch,diffusers,st,google]" @ ${HF_INFERENCE_TOOLKIT_URL} --no-cache-dir
5046

5147
ENV HF_HUB_ENABLE_HF_TRANSFER="1"
5248

containers/pytorch/inference/gpu/2.3.1/transformers/4.46.0/py311/entrypoint.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ if [[ $AIP_STORAGE_URI == gs://* ]]; then
3636
# Check if requirements.txt exists and if so install dependencies
3737
if [ -f "${HF_MODEL_DIR}/requirements.txt" ]; then
3838
echo "Installing custom dependencies from ${HF_MODEL_DIR}/requirements.txt"
39-
uv pip install -r ${HF_MODEL_DIR}/requirements.txt --no-cache-dir --system
39+
pip install -r ${HF_MODEL_DIR}/requirements.txt --no-cache-dir
4040
fi
4141
fi
4242

0 commit comments

Comments
 (0)