[TESTS] Update pytorch version to 2.5.0 (#27117)

### Details: - *item1* - *...* ### Tickets: - *ticket-id*
openvinotoolkit · Oct 28, 2024 · 30a52e8 · 30a52e8
1 parent 08365f6
commit 30a52e8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/src/frontends/pytorch/src/op_table.cpp b/src/frontends/pytorch/src/op_table.cpp
@@ -748,6 +748,7 @@ const std::unordered_map<std::string, CreatorFunction> get_supported_ops_fx() {
         {"aten._native_batch_norm_legit.no_stats", op::translate_batch_norm_legit_no_stats_fx},
         {"aten._native_batch_norm_legit_functional.default", op::translate_batch_norm_legit_fx},
         {"aten._native_batch_norm_legit_no_training.default", op::translate_batch_norm_legit_no_training_fx},
+        {"aten._safe_softmax.default", op::translate_softmax_fx},
         {"aten._scaled_dot_product_flash_attention.default", op::translate_scaled_dot_product_attention_fx},
         {"aten._scaled_dot_product_flash_attention_for_cpu.default", op::translate_scaled_dot_product_attention_fx},
         {"aten._softmax.default", op::translate_softmax_fx},

diff --git a/tests/constraints.txt b/tests/constraints.txt
@@ -28,5 +28,5 @@ networkx<=3.3
 flax<=0.10.0
 
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch~=2.4.1; platform_system != "Darwin" or platform_machine != "x86_64"
+torch~=2.5.0; platform_system != "Darwin" or platform_machine != "x86_64"
 torch~=2.2.0; platform_system == "Darwin" and platform_machine == "x86_64"
diff --git a/tests/requirements_pytorch b/tests/requirements_pytorch
@@ -3,12 +3,14 @@
 # optimum still requires numpy<2.0.0
 numpy==1.26.4; python_version < "3.12"
 numpy==2.1.1; python_version >= "3.12"
-torch==2.4.1; platform_system != "Darwin" or platform_machine != "x86_64"
+torch==2.5.0; platform_system != "Darwin" or platform_machine != "x86_64"
 torch==2.2.2; platform_system == "Darwin" and platform_machine == "x86_64"
 --extra-index-url https://download.pytorch.org/whl/cpu
 
-torchvision==0.19.1; platform_system != "Darwin" or platform_machine != "x86_64"
+torchvision==0.20.0; platform_system != "Darwin" or platform_machine != "x86_64"
 torchvision==0.17.2; platform_system == "Darwin" and platform_machine == "x86_64"
+torchaudio==2.5.0; platform_system != "Darwin" or platform_machine != "x86_64"
+torchaudio==2.2.2; platform_system == "Darwin" and platform_machine == "x86_64"
 # transformers 4.45.1 is available
 # but optimum still requires <4.45.0
 transformers==4.44.2
@@ -33,8 +35,6 @@ sentencepiece==0.2.0
 soundfile==0.12.1
 super-image==0.1.7; python_version < "3.12"
 timm==1.0.11
-torchaudio==2.4.1; platform_system != "Darwin" or platform_machine != "x86_64"
-torchaudio==2.2.2; platform_system == "Darwin" and platform_machine == "x86_64"
 wheel==0.44.0
 PyYAML==6.0.2
 kornia==0.7.3