LeapLabTHU / Agent-Attention

Official repository of Agent Attention (ECCV2024)
473 stars 35 forks source link

attention mask #32

Open Syk-yr opened 6 months ago

Syk-yr commented 6 months ago

您好,感谢您提供的思路,我想请问下,如何在agent attention 中加attention mask来屏蔽部分输入?

tian-qing001 commented 5 months ago

Hi @Syk-yr, like all other linear attention designs, agent attention cannot support arbitrary attention mask. What type of attention mask would you like to use?