wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-14 04:22:31 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Add a supports_fp64 function. (#13368)

Browse Source
This commit is contained in:
comfyanonymous 2026-04-11 18:06:36 -07:00 committed by GitHub
parent a2840e7552
commit 55ebd287ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/flux/math.py
View File

@ -16,7 +16,7 @@ def attention(q: Tensor, k: Tensor, v: Tensor, pe: Tensor, mask=None, transforme
def rope(pos: Tensor, dim: int, theta: int) -> Tensor: def rope(pos: Tensor, dim: int, theta: int) -> Tensor:
assert dim % 2 == 0 assert dim % 2 == 0
if comfy.model_management.is_device_mps(pos.device) or comfy.model_management.is_intel_xpu() or comfy.model_management.is_directml_enabled(): if not comfy.model_management.supports_fp64(pos.device):
device = torch.device("cpu") device = torch.device("cpu")
else: else:
device = pos.device device = pos.device

15
comfy/model_management.py
View File

@ -1732,6 +1732,21 @@ def supports_mxfp8_compute(device=None):
return True return True
def supports_fp64(device=None):
if is_device_mps(device):
return False
if is_intel_xpu():
return False
if is_directml_enabled():
return False
if is_ixuca():
return False
return True
def extended_fp16_support(): def extended_fp16_support():
# TODO: check why some models work with fp16 on newer torch versions but not on older # TODO: check why some models work with fp16 on newer torch versions but not on older
if torch_version_numeric < (2, 7): if torch_version_numeric < (2, 7):