diff --git a/trainium-inferentia/requirements.txt b/trainium-inferentia/requirements.txt index c378813..4e991c6 100644 --- a/trainium-inferentia/requirements.txt +++ b/trainium-inferentia/requirements.txt @@ -1,3 +1,3 @@ datasets -transformers==4.36.2 +transformers==4.38.0 optimum-neuron==0.0.21 diff --git a/trainium-inferentia/src/requirements.txt b/trainium-inferentia/src/requirements.txt index cd81490..34ea4bc 100644 --- a/trainium-inferentia/src/requirements.txt +++ b/trainium-inferentia/src/requirements.txt @@ -1,7 +1,7 @@ --extra-index-url https://pip.repos.neuron.amazonaws.com evaluate==0.4.1 accelerate==0.23.0 -transformers==4.36.2 +transformers==4.38.0 optimum-neuron==0.0.21 neuronx-distributed==0.7.0 transformers-neuronx==0.10.0.21 \ No newline at end of file