diff --git a/requirements.txt b/requirements.txt
index 690807fa..635b5a12 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,10 +9,10 @@ psutil==5.9.8
 tk==0.1.0
 customtkinter==5.2.2
 pillow==11.1.0
-torch==2.5.1; sys_platform != 'darwin'
+torch==2.0.1+cu118; sys_platform != 'darwin'
 torch==2.0.1; sys_platform == 'darwin'
-torchvision==torchvision==0.20.1; sys_platform != 'darwin'
-torchvision==0.20.1; sys_platform == 'darwin'
+torchvision==0.15.2+cu121; sys_platform != 'darwin'
+torchvision==0.15.2; sys_platform == 'darwin'
 onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
 onnxruntime-gpu==1.16.3; sys_platform != 'darwin'
 tensorflow==2.12.1; sys_platform != 'darwin'