ryan/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Actions 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 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): 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) return weight.to(device=device, dtype=dtype, non_blocking=non_blocking)
def cast_to_input(weight, input, non_blocking=False): def cast_to_input(weight, input, non_blocking=False):