Hardcode optimal compute_unit order

2023-10-23 15:10:35 +02:00
parent ef7a14750c
commit e2a99fd101
1 changed files with 6 additions and 1 deletions
@@ -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,
+                                 ],)
            }
        }