From c08b869eb7ba55330b37ef5541b293eda99c4a83 Mon Sep 17 00:00:00 2001
From: Bingsu <ks2515@naver.com>
Date: Thu, 22 Jun 2023 11:24:38 +0900
Subject: [PATCH] feat: 'use_cpu' device

---
 adetailer/__version__.py |  2 +-
 scripts/!adetailer.py    | 11 ++++++-----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/adetailer/__version__.py b/adetailer/__version__.py
index 08cd497..9fc80ec 100644
--- a/adetailer/__version__.py
+++ b/adetailer/__version__.py
@@ -1 +1 @@
-__version__ = "23.6.3"
+__version__ = "23.6.4.dev0"
diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
index 6e00e5b..29ac7d2 100644
--- a/scripts/!adetailer.py
+++ b/scripts/!adetailer.py
@@ -201,15 +201,16 @@ class AfterDetailerScript(scripts.Script):
 
     @staticmethod
     def get_ultralytics_device() -> str:
-        '`device = ""` means autodetect'
-        device = ""
+        if "adetailer" in shared.cmd_opts.use_cpu:
+            return "cpu"
+
         if platform.system() == "Darwin":
-            return device
+            return ""
 
         if any(getattr(cmd_opts, vram, False) for vram in ["lowvram", "medvram"]):
-            device = "cpu"
+            return "cpu"
 
-        return device
+        return ""
 
     def prompt_blank_replacement(
         self, all_prompts: list[str], i: int, default: str