From 2d4756431aa7ff6142f22d32052eae386b3c5238 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 25 May 2026 16:52:07 +0300 Subject: [PATCH] Update pid.py --- comfy/ldm/pixeldit/pid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/pixeldit/pid.py b/comfy/ldm/pixeldit/pid.py index ceb601647..f95698d68 100644 --- a/comfy/ldm/pixeldit/pid.py +++ b/comfy/ldm/pixeldit/pid.py @@ -251,7 +251,7 @@ class PidNet(PixDiT_T2I): pos_img = self._fetch_patch_pos(Hs, Ws, x.device, x.dtype) x_patches = F.unfold(x, kernel_size=self.patch_size, stride=self.patch_size).transpose(1, 2) - t_emb = self.t_embedder(timesteps.view(-1)).view(B, -1, self.hidden_size) + t_emb = self.t_embedder(timesteps.view(-1), x.dtype).view(B, -1, self.hidden_size) Ltxt = min(context.shape[1], self.txt_max_length) y = context[:, :Ltxt, :]