Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[pull] main from NVIDIA:main #46

Merged
merged 2 commits into from
Jan 8, 2025
Merged

[pull] main from NVIDIA:main #46

merged 2 commits into from
Jan 8, 2025

Conversation

pull[bot]
Copy link

@pull pull bot commented Jan 8, 2025

See Commits and Changes for more details.


Created by pull[bot] (v2.0.0-alpha.1)

Can you help keep this open source service alive? 💖 Please sponsor : )

zlsh80826 and others added 2 commits January 8, 2025 08:31
* Fix SWA mask for THD and forcing seqlen_kv >= seqlen_q for SWA

Signed-off-by: Reese Wang <[email protected]>

* Generalize sliding window mask

Signed-off-by: Reese Wang <[email protected]>

* Fix pylint

Signed-off-by: Reese Wang <[email protected]>

---------

Signed-off-by: Reese Wang <[email protected]>
the current implementation would release the output of ln, leading to an error if setting `return_layernorm_output=True`.

Signed-off-by: Liyuan Liu <[email protected]>
Co-authored-by: Tim Moon <[email protected]>
@pull pull bot added the ⤵️ pull label Jan 8, 2025
@pull pull bot merged commit 61cf102 into phu0ngng:main Jan 8, 2025
@pull pull bot had a problem deploying to github-pages January 8, 2025 04:33 Failure
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants