1.7 CrossAttention differernces from 1.6

[rockerBOO]

I'm working on a extension that uses the cross attention to make heatmaps. In 1.6 I had it correctly getting the cross attention. But using the same extension for 1.7 is causing the attention to create very similar heatmaps across all tokens. Is there something that changes in 1.7 that may be causing something related to CrossAttention? I'm at a loss how it is different as I looked through all the changes from 1.6 to 1.7 and couldn't identify anything that has changed.

1.7

(note that only the cat, beach or sunglasses should be heatmapped here)

In 1.6 they work as expected and I get the heatmaps in the right areas

1.6

I correlate that the heatmaps across the cross attention raw from the cross attention are also all muddled too

1.7 raw heatmaps from the cross attention (beach, sunglasses, cat in order)

I'm generally at a loss of how to fix this as I don't know what changed. If anyone has an idea of where something may have changed to look into it I would appreciate it. The extension in question is sd-webui-daam if anyone wants to take a poke at it in action. Thank you.