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

Update cfz_cudnn.toggle.py

Browse Source
This commit is contained in:
patientx 2025-06-19 00:14:37 +03:00 committed by GitHub
parent 21d999a13a
commit fbaca3d17b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cfz/cfz_cudnn.toggle.py
View File

@ -23,6 +23,7 @@ class CUDNNToggleAutoPassthrough:
def toggle(self, enable_cudnn, model=None, conditioning=None, latent=None, audio=None, image=None): def toggle(self, enable_cudnn, model=None, conditioning=None, latent=None, audio=None, image=None):
torch.backends.cudnn.enabled = enable_cudnn torch.backends.cudnn.enabled = enable_cudnn
torch.backends.cudnn.benchmark = enable_cudnn
print(f"[CUDNN_TOGGLE] torch.backends.cudnn.enabled set to {enable_cudnn}") print(f"[CUDNN_TOGGLE] torch.backends.cudnn.enabled set to {enable_cudnn}")
return_tuple = (None, None, None, None, None) return_tuple = (None, None, None, None, None)