298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-15 14:09:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apparently directml supports fp16.

Browse Source
This commit is contained in:
comfyanonymous 2025-02-20 09:29:59 -05:00
parent 29d4384a75
commit 12da6ef581
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/model_management.py
View File

@ -1021,8 +1021,6 @@ def is_directml_enabled():
return False return False
def should_use_fp16(device=None, model_params=0, prioritize_performance=True, manual_cast=False): def should_use_fp16(device=None, model_params=0, prioritize_performance=True, manual_cast=False):
global directml_enabled
if device is not None: if device is not None:
if is_device_cpu(device): if is_device_cpu(device):
return False return False
@ -1033,8 +1031,8 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma
if FORCE_FP32: if FORCE_FP32:
return False return False
if directml_enabled: if is_directml_enabled():
return False return True
if (device is not None and is_device_mps(device)) or mps_mode(): if (device is not None and is_device_mps(device)) or mps_mode():
return True return True