Merge 31f437ff79 into 80de3dc32e

2024-09-18 16:03:52 +02:00
1 changed files with 3 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -12,8 +12,10 @@ torch==2.0.1+cu118; sys_platform != 'darwin'
 torch==2.2.0; sys_platform == 'darwin'
 torchvision==0.15.2+cu118; sys_platform != 'darwin'
 torchvision==0.15.2; sys_platform == 'darwin'
+onnxruntime==1.18.0; sys_platform == 'darwin' and platform_machine != 'arm64'
 onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
-onnxruntime-gpu==1.16.3; sys_platform != 'darwin'
+onnxruntime-gpu==1.18.0; sys_platform != 'darwin'
+tensorflow==2.13.0rc1; sys_platform == 'darwin'
 tensorflow==2.12.1; sys_platform != 'darwin'
 opennsfw2==0.10.2
 protobuf==4.23.2