This commit is contained in:
comfyanonymous 2025-11-25 18:27:17 -05:00 committed by GitHub
parent 759d86c7d9
commit 2d27d9757e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -506,7 +506,6 @@ class NextDiT(nn.Module):
bsz = len(x)
pH = pW = self.patch_size
device = x[0].device
dtype = x[0].dtype
if self.pad_tokens_multiple is not None:
pad_extra = (-cap_feats.shape[1]) % self.pad_tokens_multiple