From fbaca3d17b077a41fbcfb649dbfcc8a95cf3eab6 Mon Sep 17 00:00:00 2001 From: patientx Date: Thu, 19 Jun 2025 00:14:37 +0300 Subject: [PATCH] Update cfz_cudnn.toggle.py --- cfz/cfz_cudnn.toggle.py | 1 + 1 file changed, 1 insertion(+) diff --git a/cfz/cfz_cudnn.toggle.py b/cfz/cfz_cudnn.toggle.py index 5f1cb13d5..3c99e333c 100644 --- a/cfz/cfz_cudnn.toggle.py +++ b/cfz/cfz_cudnn.toggle.py @@ -23,6 +23,7 @@ class CUDNNToggleAutoPassthrough: def toggle(self, enable_cudnn, model=None, conditioning=None, latent=None, audio=None, image=None): torch.backends.cudnn.enabled = enable_cudnn + torch.backends.cudnn.benchmark = enable_cudnn print(f"[CUDNN_TOGGLE] torch.backends.cudnn.enabled set to {enable_cudnn}") return_tuple = (None, None, None, None, None)