feat: cpu for medvram_sdxl

This commit is contained in:
Bingsu
2023-08-25 20:43:35 +09:00
parent 36f5b3548d
commit 690ce6adbd

View File

@@ -226,7 +226,8 @@ class AfterDetailerScript(scripts.Script):
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 ""