Skip to content

Commit 515159c

Browse files
committed
pin googleauth and botocore versions
1 parent abe351f commit 515159c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@ ARG OPTIMUM_NEURON_VERSION=0.0.25
1313
ARG TRANSFORMERS_VERSION
1414
ARG DATASETS_VERSION
1515
ARG GEVENT_VERSION=24.10.3
16-
ARG GAUTH_VERSION=1.35.0
1716
ARG PYTHON=python3
1817

1918
# install Hugging Face libraries and its dependencies
@@ -24,7 +23,7 @@ RUN pip install --no-cache-dir \
2423
datasets==${DATASETS_VERSION} \
2524
optimum-neuron==${OPTIMUM_NEURON_VERSION} \
2625
peft \
27-
google-auth==${GAUTH_VERSION} \
26+
"googleauth>=1.6.3,<2" \
2827
gevent==${GEVENT_VERSION}
2928

3029
# Pin numpy to version required by neuronx-cc
@@ -37,6 +36,7 @@ RUN pip install -U \
3736
Jinja2 \
3837
mlflow \
3938
tornado \
39+
botocore==1.35.19 \
4040
"urllib3>=1.26.17,<1.27"
4141

4242
RUN apt-get update \

0 commit comments

Comments
 (0)