mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-05 19:12:41 +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)
|
r.copy_(weight, non_blocking=non_blocking)
|
||||||
|
|
||||||
if signature is not None:
|
if signature is not None:
|
||||||
|
weight._v_signature = signature
|
||||||
v_tensor.copy_(r)
|
v_tensor.copy_(r)
|
||||||
comfy_aimdo.model_vbar.vbar_unpin(weight._v)
|
comfy_aimdo.model_vbar.vbar_unpin(weight._v)
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user