Skip to content

Commit cc5370d

Browse files
committed
fix(neuronx): pin huggingface_hub version
The maximum diffusers versions accepted by optimum-neuron is incompatible with the latest huggingface_hub (0.26).
1 parent 2cd855c commit cc5370d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

huggingface/pytorch/inference/docker/2.1/py3/sdk2.20.0/Dockerfile.neuronx

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ ARG NEURONX_TOOLS_VERSION=2.19.0.0
2121
# HF ARGS
2222
ARG TRANSFORMERS_VERSION
2323
ARG DIFFUSERS_VERSION=0.28.2
24+
ARG HUGGINGFACE_HUB_VERSION=0.25.2 # For compatibility with diffusers 0.28.2
2425
ARG OPTIMUM_NEURON_VERSION=0.0.25
2526
ARG SENTENCE_TRANSFORMERS=3.0.1
2627
ARG PEFT_VERSION=0.12.0
@@ -166,6 +167,7 @@ RUN curl https://aws-dlc-licenses.s3.amazonaws.com/pytorch-1.13/license.txt -o /
166167
RUN pip install --no-cache-dir \
167168
transformers[sentencepiece,audio,vision]==${TRANSFORMERS_VERSION} \
168169
diffusers==${DIFFUSERS_VERSION} \
170+
huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
169171
datasets==${DATASETS_VERSION} \
170172
optimum-neuron==${OPTIMUM_NEURON_VERSION} \
171173
sentence_transformers==${SENTENCE_TRANSFORMERS} \

0 commit comments

Comments
 (0)