QianCyrusandHongsheng Liu 8ef77be66e [Model] Avoid repeated attention and MLP work in Klein TP (#8553) ...

Signed-off-by: QianCyrus <101633534+QianCyrus@users.noreply.github.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>

2026-10-07 21:21:05 +08:00

..

__init__.py

[Model] add flux2 klein (#809)

2026-01-16 16:32:28 +08:00

flux2_klein_transformer.py

[Model] Avoid repeated attention and MLP work in Klein TP (#8553)

2026-10-07 21:21:05 +08:00

pipeline_flux2_klein.py

[Model] Avoid repeated attention and MLP work in Klein TP (#8553)

2026-10-07 21:21:05 +08:00