From 2340099d93305bfdf4eaa29e9f8d32ec92d3035f Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Thu, 6 Aug 2026 21:26:16 -0700 Subject: [PATCH] Fix full offload on minimax audio vae. (#15377) --- comfy/ldm/minimax/audio_vae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/ldm/minimax/audio_vae.py b/comfy/ldm/minimax/audio_vae.py index a1be63d54..0b9f2b6fc 100644 --- a/comfy/ldm/minimax/audio_vae.py +++ b/comfy/ldm/minimax/audio_vae.py @@ -99,7 +99,7 @@ class UpSample1d(nn.Module): def forward(self, x): _, C, _ = x.shape x = F.pad(x, (self.pad, self.pad), mode="replicate") - x = F.conv_transpose1d(x, self.filter.expand(C, -1, -1).to(x.dtype), stride=self.stride, groups=C).mul_(self.ratio) + x = F.conv_transpose1d(x, comfy.ops.cast_to_input(self.filter.expand(C, -1, -1), x), stride=self.stride, groups=C).mul_(self.ratio) x = x[..., self.pad_left:-self.pad_right] return x @@ -115,7 +115,7 @@ class LowPassFilter1d(nn.Module): def forward(self, x): _, C, _ = x.shape x = F.pad(x, (self.pad_left, self.pad_right), mode="replicate") - return F.conv1d(x, self.filter.expand(C, -1, -1).to(x.dtype), stride=self.stride, groups=C) + return F.conv1d(x, comfy.ops.cast_to_input(self.filter.expand(C, -1, -1), x), stride=self.stride, groups=C) class DownSample1d(nn.Module):