Miss device type for option --medvram

2022-09-07 18:09:30 +02:00 · 2022-09-07 18:09:30 +02:00 · aaeeef82fa
commit aaeeef82fa
parent 65fbefd033
1 changed files with 6 additions and 3 deletions
--- a/modules/lowvram.py
+++ b/modules/lowvram.py
@ -2,9 +2,12 @@ import torch

 module_in_gpu = None
 cpu = torch.device("cpu")
-gpu = torch.device("cuda")
-device = gpu if torch.cuda.is_available() else cpu
-
+if torch.has_cuda:
+    device = gpu = torch.device("cuda")
+elif torch.has_mps:
+    device = gpu = torch.device("mps")
+else:
+    device = gpu = torch.device("cpu")

 def setup_for_low_vram(sd_model, use_medvram):
    parents = {}