File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
huggingface/pytorch/training/docker/2.1/py3/sdk2.20.0 Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -13,7 +13,6 @@ ARG OPTIMUM_NEURON_VERSION=0.0.25
1313ARG TRANSFORMERS_VERSION
1414ARG DATASETS_VERSION
1515ARG GEVENT_VERSION=24.10.3
16- ARG GAUTH_VERSION=1.35.0
1716ARG 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
4242RUN apt-get update \
You can’t perform that action at this time.
0 commit comments