castorini/daam

Showing <|startoftext|> and <|endoftext|>

jiuntian opened this issue · 1 comments

jiuntian commented

Hi,

Thanks for a great work in visualising the cross attention of diffusion models. Is it possible to visualise the cross-attention map for the tokens of <|startoftext|> and <|endoftext|>? Since they play an important role in diffusion models too.

daemon commented

Good point, I'll get around to adding that.

  • 👍1