diff --git a/src/frontends/pytorch/src/op_table.cpp b/src/frontends/pytorch/src/op_table.cpp index 66c76e33032ef6..3fea4b4d595c13 100644 --- a/src/frontends/pytorch/src/op_table.cpp +++ b/src/frontends/pytorch/src/op_table.cpp @@ -746,6 +746,7 @@ const std::unordered_map 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 index 775d3287c061a1..d78e285b43bd96 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -28,5 +28,5 @@ networkx<=3.3 flax<=0.9.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 index d4e59c40f56f61..c2873210003b7d 100644 --- 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