From 05b6b36bb0ea36d72014cd5c234a7e2aca7b21fb Mon Sep 17 00:00:00 2001 From: RaINi_ Date: Mon, 19 Feb 2024 10:40:02 +0100 Subject: [PATCH] fix(model_management.py): enabled enable_math_sdp instead of enable_flash_sdp --- 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 761ff699f..b99cbf481 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -263,8 +263,8 @@ try: torch.backends.cudnn.enabled = False print("Disabling cuDNN because ZLUDA does currently not support it.") - torch.backends.cuda.enable_flash_sdp(True) - torch.backends.cuda.enable_math_sdp(False) + torch.backends.cuda.enable_flash_sdp(False) + torch.backends.cuda.enable_math_sdp(True) torch.backends.cuda.enable_mem_efficient_sdp(False) if ENABLE_PYTORCH_ATTENTION: