Update requirements_pytorch

openvinotoolkit · Oct 18, 2024 · 23a9489 · 23a9489
1 parent 2132ab6
commit 23a9489
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/requirements_pytorch b/tests/requirements_pytorch
@@ -9,6 +9,8 @@ torch==2.2.2; platform_system == "Darwin" and 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.8
-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