[cogview4][feat] Support attention mechanism with variable-length support and...

[cogview4][feat] Support attention mechanism with variable-length support and batch packing (#11349) * [cogview4] Enhance attention mechanism with variable-length support and batch packing --------- Co-authored-by: YiYi Xu <yixu310@gmail.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

[cogview4][feat] Support attention mechanism with variable-length support and...
[cogview4][feat] Support attention mechanism with variable-length support and batch packing (#11349) * [cogview4] Enhance attention mechanism with variable-length support and batch packing --------- Co-authored-by: YiYi Xu <yixu310@gmail.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
0434db9a · OleehyO · GitHub · aff574fb · 0434db9a
Unverified Commit 0434db9a authored Apr 22, 2025 by OleehyO Committed by GitHub Apr 21, 2025
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 316 additions and 24 deletions

src/diffusers/models/transformers/transformer_cogview4.py src/diffusers/models/transformers/transformer_cogview4.py +316 -24

No files found.
--- a/src/diffusers/models/transformers/transformer_cogview4.py
+++ b/src/diffusers/models/transformers/transformer_cogview4.py