Let hunyuan dit work with all prompt lengths.

Author: comfyanonymous

comfy/ldm/hydit/poolers.py

@@ -16,6 +16,7 @@ def __init__(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim:
         self.embed_dim = embed_dim
 
     def forward(self, x):
+        x = x[:,:self.positional_embedding.shape[0] - 1]
         x = x.permute(1, 0, 2)  # NLC -> LNC
         x = torch.cat([x.mean(dim=0, keepdim=True), x], dim=0)  # (L+1)NC
         x = x + self.positional_embedding[:, None, :].to(dtype=x.dtype, device=x.device)  # (L+1)NC