Alex Brooks 68a1d23e4d [Bugfix/Feature] Remove Hardcoded Flash Attention in Bagel & Support GQA in SDPA Backend (#3728) ...

Signed-off-by: Alex Brooks <albrooks@redhat.com>

2026-05-29 10:36:15 +08:00

..

__init__.py

[New Model]Bagel model(Diffusion Only) (#319)

2025-12-31 13:47:35 +08:00

autoencoder.py

[New Model]Bagel model(Diffusion Only) (#319)

2025-12-31 13:47:35 +08:00

bagel_transformer.py

[Bugfix/Feature] Remove Hardcoded Flash Attention in Bagel & Support GQA in SDPA Backend (#3728)

2026-05-29 10:36:15 +08:00

pipeline_bagel.py

[Bugfix/Feature] Remove Hardcoded Flash Attention in Bagel & Support GQA in SDPA Backend (#3728)

2026-05-29 10:36:15 +08:00