From 74e5afcbf2e520e17d04b895f56ffa73d22d3c7f Mon Sep 17 00:00:00 2001 From: Eugene Fairley Date: Wed, 9 Jul 2025 16:17:42 -0700 Subject: [PATCH] Fixes --- comfy/model_base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/comfy/model_base.py b/comfy/model_base.py index dc49c61ca..132dd5613 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -168,6 +168,7 @@ class BaseModel(torch.nn.Module): if self.manual_cast_dtype is not None: dtype = self.manual_cast_dtype + xc = xc.to(dtype) t = self.model_sampling.timestep(t).float() if context is not None: context = context.to(dtype) @@ -1208,7 +1209,7 @@ class WAN21(BaseModel): if image.shape[1] > (extra_channels - 4): image = image[:, :(extra_channels - 4)] - mask = kwargs.get("concat_mask", kwargs.get("denoise_mask", None)) if "concat_mask" in kwargs or "denoise_mask" in kwargs else None + mask = kwargs.get("concat_mask", kwargs.get("denoise_mask", None)) if mask is None: mask = torch.zeros_like(noise)[:, :4] else: