From 8cb9457e3a3f22b30dfaf7d7adb329b39a08124b Mon Sep 17 00:00:00 2001 From: Nikita Malinin Date: Tue, 16 Apr 2024 13:27:40 +0200 Subject: [PATCH] Bump transformers version (#2627) ### Changes - Bumped **transformers** package version to 4.38.2; - Updating to 4.39.3 is not possible due to **optimum** incompatibility; - Bumped **accelerate** package due to issue in **transformers** package; ### Reason for changes - Package security issues; ### Related tickets - N/A ### Tests - post_training_quantization/355/ - passed - additional post_training_quantization/358/ - passed --- tests/post_training/requirements.txt | 2 +- tests/torch/models_hub_test/requirements.txt | 2 +- tests/torch/requirements.txt | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) 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