From 73332160c8c9843876680fb04f037793c73d55b6 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sun, 18 Aug 2024 10:29:33 -0400 Subject: [PATCH] Enable non blocking transfers in lowvram mode. --- comfy/ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ops.py b/comfy/ops.py index 47e8d7a..e11d46f 100644 --- a/comfy/ops.py +++ b/comfy/ops.py @@ -34,7 +34,7 @@ def cast_bias_weight(s, input=None, dtype=None, device=None): device = input.device bias = None - non_blocking = comfy.model_management.device_should_use_non_blocking(device) + non_blocking = comfy.model_management.device_supports_non_blocking(device) if s.bias is not None: bias = cast_to(s.bias, dtype, device, non_blocking=non_blocking) if s.bias_function is not None: