huggingface / huggingface/diffusers

Can we pass output_attentions=True to DiT model such as pixart to get attention output?

Open
#9,205 1 comment 0 reactions 0 assignees View on GitHub
stale
Dominant language
Python
Stars
34.5k
Forks
7.3k
Avg merge
3d 3h
Merged PRs (30d)
91

Description

Can we pass output_attentions=True to DiT model such as pixart to get attention output? Like using output_attentions=True in transformer?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.