Skip to content

Commit 6730f3e

Browse files
Disable non blocking.
It fixed some perf issues but caused other issues that need to be debugged.
1 parent 7333216 commit 6730f3e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

comfy/ops.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222

2323
def cast_to(weight, dtype=None, device=None, non_blocking=False):
24+
non_blocking = False
2425
return weight.to(device=device, dtype=dtype, non_blocking=non_blocking)
2526

2627
def cast_to_input(weight, input, non_blocking=False):

0 commit comments

Comments
 (0)