diff --git a/comfy/ops.py b/comfy/ops.py index e11d46f9c..d76138c92 100644 --- a/comfy/ops.py +++ b/comfy/ops.py @@ -21,6 +21,7 @@ import comfy.model_management def cast_to(weight, dtype=None, device=None, non_blocking=False): + non_blocking = False return weight.to(device=device, dtype=dtype, non_blocking=non_blocking) def cast_to_input(weight, input, non_blocking=False):