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

Fix cast issue.

Browse Source
This commit is contained in:
comfyanonymous 2026-07-03 01:37:29 -04:00 committed by GitHub
parent e06033fb97
commit d3f5ae56b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/seedvr/model.py
View File

@ -22,6 +22,7 @@ from comfy.ldm.seedvr.constants import (
SEEDVR2_ROPE_PARTIAL_CHUNK_TOKENS,
)
import comfy.model_management
import comfy.ops
class Cache:
def __init__(self, disable=False, prefix="", cache=None):
@ -1041,9 +1042,12 @@ class AdaSingle(nn.Module):
)
if mode == "in":
shiftB = comfy.ops.cast_to_input(shiftB, hid)
scaleB = comfy.ops.cast_to_input(scaleB, hid)
return hid.mul_(scaleA + scaleB).add_(shiftA + shiftB)
if mode == "out":
if gateB is not None:
gateB = comfy.ops.cast_to_input(gateB, hid)
return hid.mul_(gateA + gateB)
else:
return hid.mul_(gateA)