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

Merge 31b27afe69 into 4e6a1b66a9

Browse Source
This commit is contained in:
JettHu 2026-01-25 04:05:21 +08:00 committed by GitHub
commit f9f8d6d208
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -378,7 +378,7 @@ except:
pass 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_math_sdp(True)
torch.backends.cuda.enable_flash_sdp(True) torch.backends.cuda.enable_flash_sdp(True)
torch.backends.cuda.enable_mem_efficient_sdp(True) torch.backends.cuda.enable_mem_efficient_sdp(True)