298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 10:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue with OSX.

Browse Source
This commit is contained in:
comfyanonymous 2023-07-04 02:09:02 -04:00
parent c02f3baeaf
commit 8d694cc450
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy/model_management.py
View File

@ -456,7 +456,13 @@ def mps_mode():
def is_device_cpu(device): def is_device_cpu(device):
if hasattr(device, 'type'): if hasattr(device, 'type'):
if (device.type == 'cpu' or device.type == 'mps'): if (device.type == 'cpu'):
return True
return False
def is_device_mps(device):
if hasattr(device, 'type'):
if (device.type == 'mps'):
return True return True
return False return False
@ -468,7 +474,7 @@ def should_use_fp16(device=None, model_params=0):
return True return True
if device is not None: #TODO if device is not None: #TODO
if is_device_cpu(device): if is_device_cpu(device) or is_device_mps(device):
return False return False
if FORCE_FP32: if FORCE_FP32: