File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
huggingface/pytorch/inference/docker/1.13/py3/sdk2.16.1 Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -24,6 +24,7 @@ ARG TRANSFORMERS_VERSION
24
24
ARG DIFFUSERS_VERSION=0.25.1
25
25
ARG OPTIMUM_NEURON_VERSION=0.0.18
26
26
ARG SENTENCE_TRANSFORMERS=2.2.2
27
+ ARG HUGGINGFACE_HUB_VERSION=0.25.2
27
28
28
29
# See http://bugs.python.org/issue19846
29
30
ENV LANG C.UTF-8
@@ -62,9 +63,9 @@ RUN apt-get update \
62
63
zlib1g-dev \
63
64
libcap-dev \
64
65
gpg-agent \
66
+ libarchive13 \
65
67
expat \
66
68
libexpat1 \
67
- libarchive13 \
68
69
&& rm -rf /var/lib/apt/lists/* \
69
70
&& rm -rf /tmp/tmp* \
70
71
&& apt-get clean
@@ -176,6 +177,7 @@ RUN pip install --no-cache-dir \
176
177
transformers[sentencepiece,audio,vision]==${TRANSFORMERS_VERSION} \
177
178
diffusers==${DIFFUSERS_VERSION} \
178
179
optimum-neuron==${OPTIMUM_NEURON_VERSION} \
180
+ huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
179
181
sentence_transformers==${SENTENCE_TRANSFORMERS} \
180
182
"sagemaker-huggingface-inference-toolkit<3"
181
183
You can’t perform that action at this time.
0 commit comments