Skip to content

Commit e68820d

Browse files
malav-shastrimalavhs
andauthored
patch HF PR Inference 1.13 neuronx (#4363)
* fix: patch HF PT Inference neuronx latest * revert dlc_developer_config.toml * patch libarchive13 * fix integ tests --------- Co-authored-by: malavhs <[email protected]>
1 parent e5de0c2 commit e68820d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

huggingface/pytorch/inference/docker/1.13/py3/sdk2.16.1/Dockerfile.neuronx

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ ARG TRANSFORMERS_VERSION
2424
ARG DIFFUSERS_VERSION=0.25.1
2525
ARG OPTIMUM_NEURON_VERSION=0.0.18
2626
ARG SENTENCE_TRANSFORMERS=2.2.2
27+
ARG HUGGINGFACE_HUB_VERSION=0.25.2
2728

2829
# See http://bugs.python.org/issue19846
2930
ENV LANG C.UTF-8
@@ -62,9 +63,9 @@ RUN apt-get update \
6263
zlib1g-dev \
6364
libcap-dev \
6465
gpg-agent \
66+
libarchive13 \
6567
expat \
6668
libexpat1 \
67-
libarchive13 \
6869
&& rm -rf /var/lib/apt/lists/* \
6970
&& rm -rf /tmp/tmp* \
7071
&& apt-get clean
@@ -176,6 +177,7 @@ RUN pip install --no-cache-dir \
176177
transformers[sentencepiece,audio,vision]==${TRANSFORMERS_VERSION} \
177178
diffusers==${DIFFUSERS_VERSION} \
178179
optimum-neuron==${OPTIMUM_NEURON_VERSION} \
180+
huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
179181
sentence_transformers==${SENTENCE_TRANSFORMERS} \
180182
"sagemaker-huggingface-inference-toolkit<3"
181183

0 commit comments

Comments
 (0)