wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-21 03:40:16 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

fixes to use pytorch-attention

Browse Source
This commit is contained in:
patientx 2025-02-13 19:17:35 +03:00 committed by GitHub
parent f9ee02080f
commit bce4176d3d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/model_management.py
View File

@ -237,10 +237,10 @@ try:
except: except:
pass pass
if ENABLE_PYTORCH_ATTENTION: #if ENABLE_PYTORCH_ATTENTION:
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)
try: try:
if is_nvidia() and args.fast: if is_nvidia() and args.fast: