Skip to content

Commit 740f2ee

Browse files
committed
added debug for values
1 parent 7066e2f commit 740f2ee

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

src/maxdiffusion/pipelines/wan/wan_pipeline.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,6 +393,7 @@ def _get_t5_prompt_embeds(
393393
)
394394
text_input_ids, mask = text_inputs.input_ids, text_inputs.attention_mask
395395
seq_lens = mask.gt(0).sum(dim=1).long()
396+
print("Seq lens:", seq_lens)
396397
prompt_embeds = self.text_encoder(text_input_ids, mask).last_hidden_state
397398
prompt_embeds = [u[:v] for u, v in zip(prompt_embeds, seq_lens)]
398399
prompt_embeds = torch.stack(

0 commit comments

Comments
 (0)