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

mix_ascend_bf16_infer_err

Browse Source
This commit is contained in:
zhoufan2956 2025-02-12 17:13:10 +08:00
parent d9f0fcdb0c
commit a321f9ddf3
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_management.py
View File

@ -1083,6 +1083,9 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
if is_intel_xpu(): if is_intel_xpu():
return True return True
if is_ascend_npu():
return True
props = torch.cuda.get_device_properties(device) props = torch.cuda.get_device_properties(device)
if props.major >= 8: if props.major >= 8:
return True return True