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)