promptbench Want to know what code is used to implement the attention visualization in the figure below?

Want to know what code is used to implement the attention visualization in the figure below?

Open zl-comment opened this issue 1 year ago • 2 comments

微信图片_20240913235257

Sep 13 '24 15:09 zl-comment

Hi, you could refer to #78, the code is in this file.

Sep 13 '24 15:09 kaijiezhu11

Thanks, I already know how to run it, thank you very much

Sep 13 '24 16:09 zl-comment

Stale issue message

Nov 13 '24 06:11 github-actions[bot]

When I run the code. It meets a problem. ValueError: Expected input batch_size (22) to match target batch_size (1).

Jul 13 '25 04:07 ZJunBo