jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Disable non blocking.

Browse Source 
It fixed some perf issues but caused other issues that need to be debugged.
This commit is contained in:
comfyanonymous
2024-08-18 14:38:09 -04:00
committed by GitHub
parent 73332160c8
commit 6730f3e1a3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ops.py
View File

@@ -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):