diff --git a/huggingface/pytorch/training/docker/2.1/py3/sdk2.20.0/Dockerfile.neuronx b/huggingface/pytorch/training/docker/2.1/py3/sdk2.20.0/Dockerfile.neuronx index eb6d062e11c3..ad5b96f272dd 100644 --- a/huggingface/pytorch/training/docker/2.1/py3/sdk2.20.0/Dockerfile.neuronx +++ b/huggingface/pytorch/training/docker/2.1/py3/sdk2.20.0/Dockerfile.neuronx @@ -23,7 +23,7 @@ RUN pip install --no-cache-dir \ datasets==${DATASETS_VERSION} \ optimum-neuron==${OPTIMUM_NEURON_VERSION} \ peft \ - "googleauth>=1.6.3,<2" \ + "google-auth>=1.6.3,<2" \ gevent==${GEVENT_VERSION} # Pin numpy to version required by neuronx-cc