huggingface / huggingface/diffusers

[core] refactor `attention_processor.py` the easy way

オープン
#10,022 コメント 6 件 リアクション 0 件 担当者 1 名 @DN6 が担当を希望しています GitHub で見る
wip
主要言語
Python
スター
34.5k
フォーク
7.3k
平均マージ
3日 3時間
マージ済み PR(30日)
91

説明

With @DN6 we have been discussing an idea about breaking up `src/diffusers/models/attention_processor.py` as it's getting excruciatingly longer and longer. The idea is simple and won't very likely require multiple rounds of PRs.

* Create a module named `attention_processor`.
* Split the attention processor classes with respect to the model names they are used. This makes sense because pretty much most of the models (at least the widely used ones) have their own attention processors at this point. For example, `FluxAttnProcessor2_0`, `FluxAttnProcessor2_0_NPU`, `FusedFluxAttnProcessor2_0` could go to `attention_processor/flux_attention_processors.py`.
* Other attention processors (such as `AttnProcessor`, `AttnProcessor2_0`, etc.) which are shared across different models could live in a common file i.e., `attention_processor/common.py`.

Since `attention_prcoessor/` will have an init, I don't think there will be any breaking changes.

コントリビューションガイド

コントリビューションガイドを開く

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。