wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 13:32:36 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix fp16 autocast RuntimeError on Macs.

Browse Source
This commit is contained in:
shiimizu 2023-09-03 02:16:09 -07:00
parent a74c5dbf37
commit 4b471cb8c2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_management.py
View File

@ -466,7 +466,8 @@ def vae_dtype():
def get_autocast_device(dev): def get_autocast_device(dev):
if hasattr(dev, 'type'): 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" return "cuda"