Changed to use Macbook MPS device

kohya_gui/blip2_caption_gui.py

@@ -12,8 +12,13 @@
 
 
 def load_model():
-    # Set the device to GPU if available, otherwise use CPU
-    device = "cuda" if torch.cuda.is_available() else "cpu"
+    # Set the device to GPU if available, MPS if available, otherwise use CPU
+    if torch.cuda.is_available():
+      device = torch.device("cuda") 
+    elif torch.backends.mps.is_available():
+      device = torch.device("mps")
+    else:
+      device = torch.device("cpu")
 
     # Initialize the BLIP2 processor
     processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")

kohya_gui/extract_lora_gui.py

@@ -326,6 +326,7 @@ def change_sdxl(sdxl):
                 choices=[
                     "cpu",
                     "cuda",
+                    "mps"
                 ],
                 value="cuda",
                 interactive=True,

kohya_gui/extract_lycoris_locon_gui.py

@@ -282,6 +282,7 @@ def list_save_to(path):
                 choices=[
                     "cpu",
                     "cuda",
+                    "mps"
                 ],
                 value="cuda",
                 interactive=True,