diff --git a/tests/post_training/requirements.txt b/tests/post_training/requirements.txt index 3b223aff441..24246bfb2ca 100644 --- a/tests/post_training/requirements.txt +++ b/tests/post_training/requirements.txt @@ -2,7 +2,7 @@ torch==2.2.1 --extra-index-url https://download.pytorch.org/whl/cpu torchvision==0.17.1 -transformers==4.36.0 +transformers==4.38.2 onnx==1.16.0 onnxruntime==1.17.1 pytest==8.0.2 diff --git a/tests/torch/models_hub_test/requirements.txt b/tests/torch/models_hub_test/requirements.txt index 6e5c5829757..4f1c41ddac4 100644 --- a/tests/torch/models_hub_test/requirements.txt +++ b/tests/torch/models_hub_test/requirements.txt @@ -2,7 +2,7 @@ torch==2.2.1 --extra-index-url https://download.pytorch.org/whl/cpu torchvision==0.17.1 -transformers==4.36.0 +transformers==4.38.2 pytest==8.0.2 timm==0.9.2 scikit-learn==1.2.2 diff --git a/tests/torch/requirements.txt b/tests/torch/requirements.txt index 244e037e95e..d9a365f3e34 100644 --- a/tests/torch/requirements.txt +++ b/tests/torch/requirements.txt @@ -8,8 +8,8 @@ pytest-dependency>=0.5.1 virtualenv # Required for search_building_blocks tests -transformers[torch]~=4.36.0 -accelerate==0.24.1 +transformers[torch]~=4.38.2 +accelerate==0.28.0 # Required for movement_sparsity tests datasets~=2.14.0