huggingface / huggingface/diffusers

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

Ouverte
#10,022 6 commentaires 0 réactions 1 personne assignée Réclamée par @DN6 Voir sur GitHub
wip
Langage dominant
Python
Étoiles
34.5k
Forks
7.3k
Merge moyen
3 j 3 h
PR mergées (30 j)
91

Description

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.

Guide de contribution

Ouvrir le guide de contribution

Évaluation

Cette issue n'a pas encore été évaluée.

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.