Only DoubleStreamTransformerBlock

[Dumeowmeow]

Thank you for your nice work! It helps me a lot.But I have a qustion:

attention-map-diffusers/attention_map_diffusers/modules.py

Line 2174 in d1478b6

if hasattr(self, "store_attn_map") and encoder_hidden_states is not None:

Does this mean only the attention maps of the DoubleStreamTransformerBlock can be visualized？