wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-21 23:41:28 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Merge branch 'comfyanonymous:master' into master

Browse Source
This commit is contained in:
patientx
2024-08-19 10:02:52 +03:00
committed by GitHub
parent a09513099a 39f114c44b
commit eb8d7f86d1
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/ops.py
+3 -2
View File
@@ -21,8 +21,9 @@ 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)
r = torch.empty_like(weight, dtype=dtype, device=device)
r.copy_(weight, non_blocking=non_blocking)
return r
def cast_to_input(weight, input, non_blocking=False):
return cast_to(weight, input.dtype, input.device, non_blocking=non_blocking)