From b588fdd0f5c8639fb00409aea73edbd78330373f Mon Sep 17 00:00:00 2001 From: Simon Lui <502929+simonlui@users.noreply.github.com> Date: Thu, 19 Dec 2024 10:34:23 -0800 Subject: [PATCH] Flip minor version check sign. --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 960c1994..9d8b4ccd 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -219,7 +219,7 @@ if ENABLE_PYTORCH_ATTENTION: torch.backends.cuda.enable_flash_sdp(True) torch.backends.cuda.enable_mem_efficient_sdp(True) -if int(torch_version[0]) == 2 and int(torch_version[2]) <= 5: +if int(torch_version[0]) == 2 and int(torch_version[2]) >= 5: torch.backends.cuda.allow_fp16_bf16_reduction_math_sdp(True) if args.lowvram: