From e2a99fd1018bcfeb43f81291d1112df66c647ce4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adrian=20Szczepa=C5=84ski?= <kontakt@aszczepanski.pl>
Date: Sun, 22 Oct 2023 23:27:44 +0200
Subject: [PATCH] Hardcode optimal compute_unit order

---
 loaders.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/loaders.py b/loaders.py
index b6955c9..94353a7 100644
--- a/loaders.py
+++ b/loaders.py
@@ -16,7 +16,12 @@ class CoreMLLoader:
         return {
             "required": {
                 "coreml_name": (list(s.coreml_filenames().keys()),),
-                "compute_unit": (list(ComputeUnit.__members__.keys()),)
+                "compute_unit": ([
+                                     ComputeUnit.CPU_AND_NE.name,
+                                     ComputeUnit.CPU_AND_GPU.name,
+                                     ComputeUnit.ALL.name,
+                                     ComputeUnit.CPU_ONLY.name,
+                                 ],)
             }
         }