xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 05:10:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix bfloat16 potentially not being enabled on mps.

Browse Source
This commit is contained in:
comfyanonymous 2024-08-01 16:18:14 -04:00
parent bd6f023ff6
commit f5b2fc6e6b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@ -897,7 +897,10 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
if directml_enabled:
return False
if cpu_mode() or mps_mode():
if mps_mode():
return True
if cpu_mode():
return False
if is_intel_xpu():