Merge pull request cubiq#223 from Visionatrix/fix/requirements

install correct onnxruntime on mac
maepopi · Sep 28, 2024 · 84ec3a0 · 84ec3a0
2 parents 579bdf6 + 9e35576
commit 84ec3a0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/InstantID.py b/InstantID.py
@@ -205,7 +205,7 @@ class InstantIDFaceAnalysis:
     def INPUT_TYPES(s):
         return {
             "required": {
-                "provider": (["CPU", "CUDA", "ROCM"], ),
+                "provider": (["CPU", "CUDA", "ROCM", "CoreML"], ),
             },
         }
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,4 @@
 insightface
 onnxruntime
-onnxruntime-gpu
+onnxruntime-gpu; sys_platform != 'darwin' and platform_machine == 'x86_64'
+onnxruntime-silicon; sys_platform == 'darwin'