wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 14:20:27 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Update zluda.py

Browse Source
This commit is contained in:
patientx 2025-08-24 12:01:20 +03:00 committed by GitHub
parent dba9d20791
commit c92a07594b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/customzluda/zluda.py
View File

@ -652,7 +652,8 @@ def do_hijack():
print(" :: Configuring PyTorch backends...") print(" :: Configuring PyTorch backends...")
torch.backends.cuda.enable_mem_efficient_sdp(False) torch.backends.cuda.enable_mem_efficient_sdp(False)
torch.backends.cuda.enable_mem_efficient_sdp = do_nothing torch.backends.cuda.enable_mem_efficient_sdp = do_nothing
# torch.backends.cudnn.enabled = True #this was here from the early days , always assumed this was required for other stuff, disabling it. torch.backends.cudnn.enabled = False
torch.backends.cudnn.benchmark = False
if hasattr(torch.backends.cuda, "enable_flash_sdp"): if hasattr(torch.backends.cuda, "enable_flash_sdp"):
torch.backends.cuda.enable_flash_sdp(True) torch.backends.cuda.enable_flash_sdp(True)
print(" :: Disabled CUDA flash attention") print(" :: Disabled CUDA flash attention")