Fix potential mps issue of supports_fp64 called with None. (#16136)

This commit is contained in:
comfyanonymous
2026-09-05 18:00:11 -07:00
committed by GitHub
parent 18ebc2afd4
commit d1c5e64119

View File

@@ -2008,7 +2008,7 @@ def supports_mxfp8_compute(device=None):
return True
def supports_fp64(device=None):
if is_device_mps(device):
if (device is not None and is_device_mps(device)) or mps_mode():
return False
if is_intel_xpu():