File tree 1 file changed +2
-0
lines changed
huggingface/pytorch/inference/docker/2.1/py3/sdk2.20.0 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -21,6 +21,7 @@ ARG NEURONX_TOOLS_VERSION=2.19.0.0
21
21
# HF ARGS
22
22
ARG TRANSFORMERS_VERSION
23
23
ARG DIFFUSERS_VERSION=0.28.2
24
+ ARG HUGGINGFACE_HUB_VERSION=0.25.2 # For compatibility with diffusers 0.28.2
24
25
ARG OPTIMUM_NEURON_VERSION=0.0.25
25
26
ARG SENTENCE_TRANSFORMERS=3.0.1
26
27
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 /
166
167
RUN pip install --no-cache-dir \
167
168
transformers[sentencepiece,audio,vision]==${TRANSFORMERS_VERSION} \
168
169
diffusers==${DIFFUSERS_VERSION} \
170
+ huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
169
171
datasets==${DATASETS_VERSION} \
170
172
optimum-neuron==${OPTIMUM_NEURON_VERSION} \
171
173
sentence_transformers==${SENTENCE_TRANSFORMERS} \
You can’t perform that action at this time.
0 commit comments