From 4b471cb8c2d797eee771229bada40ae5214e3247 Mon Sep 17 00:00:00 2001
From: shiimizu <shiimizu@protonmail.com>
Date: Sun, 3 Sep 2023 02:16:09 -0700
Subject: [PATCH] Fix fp16 autocast RuntimeError on Macs.

---
 comfy/model_management.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index aca8af999..06e2b5934 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -466,7 +466,8 @@ def vae_dtype():
 
 def get_autocast_device(dev):
     if hasattr(dev, 'type'):
-        return dev.type
+        return dev.type if not (dev.type == "mps" and should_use_fp16(
+            dev, prioritize_performance=False)) else "cpu"
     return "cuda"