From f0fae811d86d05e8b94f31deec8b5456525f6b7b Mon Sep 17 00:00:00 2001 From: Kenneth Estanislao Date: Sun, 29 Jun 2025 15:03:35 +0800 Subject: [PATCH] Update requirements.txt should improve the performance by 30% --- requirements.txt | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/requirements.txt b/requirements.txt index 6d9f8b8..68b4b2c 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/cu128 numpy>=1.23.5,<2 typing-extensions>=4.8.0 @@ -10,12 +10,12 @@ psutil==5.9.8 tk==0.1.0 customtkinter==5.2.2 pillow==11.1.0 -torch==2.5.1+cu118; sys_platform != 'darwin' +torch; sys_platform != 'darwin' torch==2.5.1; sys_platform == 'darwin' -torchvision==0.20.1; sys_platform != 'darwin' +torchvision; sys_platform != 'darwin' torchvision==0.20.1; sys_platform == 'darwin' -onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64' -onnxruntime-gpu==1.17; sys_platform != 'darwin' +onnxruntime-silicon==1.21.0; sys_platform == 'darwin' and platform_machine == 'arm64' +onnxruntime-gpu==1.21.0; sys_platform != 'darwin' tensorflow; sys_platform != 'darwin' opennsfw2==0.10.2 protobuf==4.23.2