ConceptAttention is an interpretability method designed for multi-modal diffusion transformers, specifically implemented for the Flux DiT architecture using PyTorch. The article provides installation instructions and a code example for generating images and concept attention heatmaps. It also references the associated research paper for further details.
interpretability ✓
diffusion-transformers ✓
+ pytorch
image-generation ✓
concept-attention ✓