mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-15 05:57:20 +00:00
Disable non blocking.
It fixed some perf issues but caused other issues that need to be debugged.
This commit is contained in:
parent
73332160c8
commit
6730f3e1a3
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user