From d3f5ae56b3a3acc2aa9d974660000651e69fac05 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Fri, 3 Jul 2026 01:37:29 -0400 Subject: [PATCH] Fix cast issue. --- comfy/ldm/seedvr/model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/comfy/ldm/seedvr/model.py b/comfy/ldm/seedvr/model.py index 772ec91c5..a978698d5 100644 --- a/comfy/ldm/seedvr/model.py +++ b/comfy/ldm/seedvr/model.py @@ -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)