A problem for extend_attention module

[dreamer121121]

Hi thanks for your release. I have a question about the extended attention module. I found that there is a little different between the code and description, one frame query only its'k not other keyframe's K