wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-27 09:27:24 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Update pid.py

Browse Source
This commit is contained in:
kijai 2026-05-25 16:52:07 +03:00
parent 5497baaf8d
commit 2d4756431a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/pixeldit/pid.py
View File

@ -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, :]