wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-13 07:40:50 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Update model_management.py

Browse Source
This commit is contained in:
patientx 2025-02-28 16:37:55 +03:00 committed by GitHub
parent 1871a594ba
commit af43425ab5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

@ -273,8 +273,8 @@ except:
pass
# if ENABLE_PYTORCH_ATTENTION:
# torch.backends.cuda.enable_math_sdp(True)
if ENABLE_PYTORCH_ATTENTION:
torch.backends.cuda.enable_math_sdp(True)
# torch.backends.cuda.enable_flash_sdp(True)
# torch.backends.cuda.enable_mem_efficient_sdp(True)