diff --git a/requirements.txt b/requirements.txt index 0ce3bf2..f9e9754 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/cu121 +--extra-index-url https://download.pytorch.org/whl/cu118 numpy>=1.23.5,<2 typing-extensions>=4.8.0 @@ -10,10 +10,10 @@ psutil==5.9.8 tk==0.1.0 customtkinter==5.2.2 pillow==11.1.0 -torch==2.5.1+cu121; sys_platform != 'darwin' -torch==2.5.1+cu121; sys_platform == 'darwin' +torch==2.5.1+cu118; sys_platform != 'darwin' +torch==2.5.1+cu118; sys_platform == 'darwin' torchvision==0.20.1; sys_platform != 'darwin' -torchvision==0.20.1+cu121; sys_platform == 'darwin' +torchvision==0.20.1+cu118; 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'