diff --git a/requirements.txt b/requirements.txt
index db81d80..bbd0f80 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
---extra-index-url https://download.pytorch.org/whl/cu118
+--extra-index-url https://download.pytorch.org/whl/nightly/cu128
 
 numpy>=1.23.5,<2
 typing-extensions>=4.8.0
@@ -12,8 +12,8 @@ customtkinter==5.2.2
 pillow==11.1.0
 torch==2.5.1+cu118; sys_platform != 'darwin'
 torch==2.5.1; sys_platform == 'darwin'
-torchvision==0.20.1; sys_platform != 'darwin'
-torchvision==0.20.1; sys_platform == 'darwin'
+torchvision; sys_platform != 'darwin'
+torchvision; sys_platform == 'darwin'
 onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
 onnxruntime-gpu==1.16.3; sys_platform != 'darwin'
 tensorflow; sys_platform != 'darwin'