diff --git a/huggingface/pytorch/inference/docker/1.13/py3/sdk2.16.1/Dockerfile.neuronx b/huggingface/pytorch/inference/docker/1.13/py3/sdk2.16.1/Dockerfile.neuronx index 51cbf4ee87b9..6ce00e81c3cf 100644 --- a/huggingface/pytorch/inference/docker/1.13/py3/sdk2.16.1/Dockerfile.neuronx +++ b/huggingface/pytorch/inference/docker/1.13/py3/sdk2.16.1/Dockerfile.neuronx @@ -24,6 +24,7 @@ ARG TRANSFORMERS_VERSION ARG DIFFUSERS_VERSION=0.25.1 ARG OPTIMUM_NEURON_VERSION=0.0.18 ARG SENTENCE_TRANSFORMERS=2.2.2 +ARG HUGGINGFACE_HUB_VERSION=0.25.2 # See http://bugs.python.org/issue19846 ENV LANG C.UTF-8 @@ -62,9 +63,9 @@ RUN apt-get update \ zlib1g-dev \ libcap-dev \ gpg-agent \ + libarchive13 \ expat \ libexpat1 \ - libarchive13 \ && rm -rf /var/lib/apt/lists/* \ && rm -rf /tmp/tmp* \ && apt-get clean @@ -176,6 +177,7 @@ RUN pip install --no-cache-dir \ transformers[sentencepiece,audio,vision]==${TRANSFORMERS_VERSION} \ diffusers==${DIFFUSERS_VERSION} \ optimum-neuron==${OPTIMUM_NEURON_VERSION} \ + huggingface_hub==${HUGGINGFACE_HUB_VERSION} \ sentence_transformers==${SENTENCE_TRANSFORMERS} \ "sagemaker-huggingface-inference-toolkit<3"