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

feat: pytorch attention does not seem to crash the driver anymore

Browse Source
This commit is contained in:
LeagueRaINi 2024-02-26 02:32:21 +01:00
parent 4b2e42e8fe
commit 0683dc912c
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management.py
View File

@ -267,10 +267,6 @@ try:
torch.backends.cuda.enable_math_sdp(True) 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:
print("Disabling pytorch cross attention because ZLUDA does currently not support it.")
ENABLE_PYTORCH_ATTENTION = False
print("Device:", torch_device_name) print("Device:", torch_device_name)
except: except:
print("Could not pick default device.") print("Could not pick default device.")