diff --git a/python/requirements/ml/requirements_train.txt b/python/requirements/ml/requirements_train.txt index 286c4ab5a929..10343511d16b 100644 --- a/python/requirements/ml/requirements_train.txt +++ b/python/requirements/ml/requirements_train.txt @@ -10,7 +10,7 @@ tensorboardX==2.4.1 # `python/ray/air/examples/huggingface/huggingface_basic_language_modeling_example.py` # `python/ray/air/tests/test_huggingface_trainer.py` transformers==4.18.0; python_version <= '3.6' -transformers==4.19.1; python_version > '3.6' +transformers==4.22.1; python_version > '3.6' accelerate==0.5.1 datasets==2.0.0 sentencepiece==0.1.96 diff --git a/python/requirements/ml/requirements_tune.txt b/python/requirements/ml/requirements_tune.txt index 8105dfb5ca8b..cec58dd3466e 100644 --- a/python/requirements/ml/requirements_tune.txt +++ b/python/requirements/ml/requirements_tune.txt @@ -36,7 +36,7 @@ scikit-optimize==0.8.1 sigopt==7.5.0 timm==0.4.5 transformers==4.18.0; python_version <= '3.6' -transformers==4.19.1; python_version > '3.6' +transformers==4.22.1; python_version > '3.6' wandb==0.12.15 xgboost==1.3.3 zoopt==0.4.1