mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-03 02:00:29 +08:00
add missing signature set for non comfy
This commit is contained in:
parent
362d1d845e
commit
fce0d31dcb
@ -1208,6 +1208,7 @@ def cast_to(weight, dtype=None, device=None, non_blocking=False, copy=False, str
|
||||
r.copy_(weight, non_blocking=non_blocking)
|
||||
|
||||
if signature is not None:
|
||||
weight._v_signature = signature
|
||||
v_tensor.copy_(r)
|
||||
comfy_aimdo.model_vbar.vbar_unpin(weight._v)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user