From af43425ab5a0fe5ecfc6d2b2021a184d15607f55 Mon Sep 17 00:00:00 2001 From: patientx Date: Fri, 28 Feb 2025 16:37:55 +0300 Subject: [PATCH] Update model_management.py --- comfy/model_management.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 19a204cb2..01a2e5677 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -273,8 +273,8 @@ except: pass -# if ENABLE_PYTORCH_ATTENTION: - # torch.backends.cuda.enable_math_sdp(True) +if ENABLE_PYTORCH_ATTENTION: + torch.backends.cuda.enable_math_sdp(True) # torch.backends.cuda.enable_flash_sdp(True) # torch.backends.cuda.enable_mem_efficient_sdp(True)