298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-13 22:31:08 +00: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 48eb1399c0
commit a6decf1e62
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():