search

v0xie / sd-webui-incantations

Enhance Stable Diffusion image quality, prompt following, and more through multiple implementations of novel algorithms for Automatic1111 WebUI.
GNU General Public License v3.0
120 stars 7 forks source link

Better Save Attention Maps #48

Closed v0xie closed 1 month ago

v0xie commented 1 month ago

Refactors the Save Attention Maps script for plotting cross attention maps Adds options for different attention map filtering (Per-Token Attention Maps, One-Hot Argmax maps) Each plot shows the token index, the token id (tokenized word), and the word associated with the token id. Adds a unique sequence number to the beginning of the saved files to prevent overwriting old files

image image

v0xie commented 1 month ago

One-hot map is not very useful as-is, but this is OK for now