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

fix(model_management.py): enabled enable_math_sdp instead of enable_flash_sdp

Browse Source
This commit is contained in:
RaINi_ 2024-02-19 10:40:02 +01:00 committed by GitHub
parent 0cd6f8a0f0
commit 05b6b36bb0
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

@ -263,8 +263,8 @@ try:
torch.backends.cudnn.enabled = False torch.backends.cudnn.enabled = False
print("Disabling cuDNN because ZLUDA does currently not support it.") print("Disabling cuDNN because ZLUDA does currently not support it.")
torch.backends.cuda.enable_flash_sdp(True) torch.backends.cuda.enable_flash_sdp(False)
torch.backends.cuda.enable_math_sdp(False) torch.backends.cuda.enable_math_sdp(True)
torch.backends.cuda.enable_mem_efficient_sdp(False) torch.backends.cuda.enable_mem_efficient_sdp(False)
if ENABLE_PYTORCH_ATTENTION: if ENABLE_PYTORCH_ATTENTION: