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

castorini / daam

Diffusion attentive attribution maps for interpreting Stable Diffusion.

MIT License

669 stars 61 forks source link

Showing <|startoftext|> and <|endoftext|> #58

Open jiuntian opened 7 months ago

jiuntian commented 7 months ago

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 7 months ago

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