xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 07:50:02 +08:00
Code Issues Packages Projects Releases Wiki 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 acc2ed8389
commit 22b99fdff4
1 changed files with 1 additions and 0 deletions
Show Stats 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):