diff --git a/requirements.txt b/requirements.txt index 70b082f..2652c0e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,8 @@ psutil==5.9.5 tk==0.1.0 pillow==9.5.0 torch==2.0.1 -onnxruntime-gpu==1.15.0 +onnxruntime==1.15.0; sys_platform == 'linux' +onnxruntime-gpu==1.15.0; sys_platform != 'darwin' tensorflow==2.12.0 opennsfw2==0.10.2 protobuf==4.23.2