Enable non blocking transfers in lowvram mode.

6 months ago · 73332160c8
parent 2622c55aff
commit 73332160c8
1 changed files with 1 additions and 1 deletions
--- 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: