Skip to content

Commit dc4f451

Browse files
committed
some debug added to t5_prompt_embeds
1 parent 7b8efca commit dc4f451

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/maxdiffusion/pipelines/wan/wan_pipeline.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -420,7 +420,8 @@ def _get_t5_prompt_embeds(
420420
max_sequence_length: int = 226,
421421
):
422422
jax.debug.print("Prompt Shape")
423-
jax.debug.print(prompt.shape)
423+
jax.debug.print(f"Length of prompt list: {len(prompt)}")
424+
424425
prompt = [prompt] if isinstance(prompt, str) else prompt
425426
prompt = [prompt_clean(u) for u in prompt]
426427
batch_size = len(prompt)

0 commit comments

Comments
 (0)