A code to get cross-attention map at each step in diffusion sampling.
use:
python get_attnmap.py
you can get cross-attention maps of each step and the average.
A code to get cross-attention map at each step in diffusion sampling.
use:
python get_attnmap.py
you can get cross-attention maps of each step and the average.