Is there a problem with Self Attention's visualization code?

[zhtjtcz]

Thanks for the nice job.

In video_diffusion/prompt_attention/visualization.py, function show_self_attention_comp does not gives prompts when it calls function aggregate_attention. This appears to cause a missing parameter error？