Fix: Omitted consideration of macOS support

2024-09-23 11:10:41 +09:00
parent f21f6435ad
commit f9607ad891
3 changed files with 16 additions and 4 deletions
--- a/requirements-infer.txt
+++ b/requirements-infer.txt
@@ -12,7 +12,8 @@ nltk<=3.8.1
 num2words
 numpy<2
 onnxruntime
-onnxruntime-gpu
+onnxruntime-directml; sys_platform == 'win32'
+onnxruntime-gpu; sys_platform != 'darwin'
 # onnxsim
 # protobuf==4.25
 psutil