wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-21 11:50:16 +08:00
Code Issues Actions 17 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
commit eb8d7f86d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/ops.py
View File

@ -21,8 +21,9 @@ 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 r = torch.empty_like(weight, dtype=dtype, device=device)
return weight.to(device=device, dtype=dtype, non_blocking=non_blocking) r.copy_(weight, non_blocking=non_blocking)
return r
def cast_to_input(weight, input, non_blocking=False): def cast_to_input(weight, input, non_blocking=False):
return cast_to(weight, input.dtype, input.device, non_blocking=non_blocking) return cast_to(weight, input.dtype, input.device, non_blocking=non_blocking)