Disable non blocking.

It fixed some perf issues but caused other issues that need to be debugged.
6 months ago · 6730f3e1a3
parent 73332160c8
commit 6730f3e1a3
1 changed files with 1 additions and 0 deletions
--- 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):