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

ChenyangQiQi / FateZero

[ICCV 2023 Oral] "FateZero: Fusing Attentions for Zero-shot Text-based Video Editing"

http://fate-zero-edit.github.io/

MIT License

1.11k stars 108 forks source link

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

Open zhtjtcz opened 7 months ago

zhtjtcz commented 7 months ago

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？