wooyeolbaek/attention-map-diffusers
A tool for extracting and visualizing cross-attention maps from text-to-image diffusion models using HuggingFace Diffusers.

Velocity · 7d
+0.4
★ / day
Trend
→steady
star history
This repository provides utilities for extracting and visualizing cross-attention maps from diffusion models. It hooks into the HuggingFace Diffusers library to capture attention weights during the diffusion process, enabling analysis of how text prompts influence image generation. The tool is compatible with Stable Diffusion 2/3, SDXL, and Flux models, supporting both batch operations and layer-wise/timestep-based attention map saving.