Skip to content

Commit

Permalink
feat: cpu for medvram_sdxl
Browse files Browse the repository at this point in the history
  • Loading branch information
Bing-su committed Aug 25, 2023
1 parent 36f5b35 commit 690ce6a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion scripts/!adetailer.py
Original file line number Diff line number Diff line change
Expand Up @@ -226,7 +226,8 @@ def get_ultralytics_device() -> str:
if platform.system() == "Darwin":
return ""

if any(getattr(cmd_opts, vram, False) for vram in ["lowvram", "medvram"]):
vram_args = ["lowvram", "medvram", "medvram_sdxl"]
if any(getattr(cmd_opts, vram, False) for vram in vram_args):
return "cpu"

return ""
Expand Down

0 comments on commit 690ce6a

Please sign in to comment.