RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Only enable torch.backends.cuda sdp(s) when not cpu mode

This commit is contained in:
JettHu 2025-07-01 11:19:13 +08:00
parent c46268bf60
commit 31b27afe69
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -317,7 +317,7 @@ except:
pass
if ENABLE_PYTORCH_ATTENTION:
if ENABLE_PYTORCH_ATTENTION and cpu_state != CPUState.CPU:
torch.backends.cuda.enable_math_sdp(True)
torch.backends.cuda.enable_flash_sdp(True)
torch.backends.cuda.enable_mem_efficient_sdp(True)