From 4d2aea37b73aa4da47ed49ae0a3158ec88330852 Mon Sep 17 00:00:00 2001 From: Kenneth Estanislao Date: Thu, 6 Feb 2025 00:43:20 +0800 Subject: [PATCH] Update requirements.txt --- requirements.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 4e203f72..90d25cc9 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.5.1; sys_platform == 'darwin' +torch==2.5.1+cu121; sys_platform != 'darwin' +torch==2.5.1+cu121; sys_platform == 'darwin' torchvision==0.20.1; sys_platform != 'darwin' -torchvision==0.20.1; sys_platform == 'darwin' +torchvision==0.20.1+cu121; 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'