Skip to content

Conversation

@stas00
Copy link
Collaborator

@stas00 stas00 commented Oct 20, 2025

Ulysses/ALST integration with HF Accelerate:

  • Allow UlyssesSPAttentionHF.register_with_transformers to get a model obj as an argument, to match HF accelerate's workflow
  • Fix existing Ulysses' tests to tests z2 instead of z1
  • Improve documentation
  • Add a defensive check

The HF Accelerate PR that depends on this PR is here huggingface/accelerate#3817

Signed-off-by: Stas Bekman <[email protected]>
Signed-off-by: Stas Bekman <[email protected]>
Signed-off-by: Stas Bekman <[email protected]>
@stas00 stas00 marked this pull request as ready for review October 22, 2025 17:16
Signed-off-by: Stas Bekman <[email protected]>
@stas00 stas00 enabled auto-merge (squash) October 22, 2025 17:37
@stas00 stas00 merged commit 64c0052 into master Oct 22, 2025
12 checks passed
@stas00 stas00 deleted the stas/alst-hf-accelerate branch October 22, 2025 18:57
@stas00
Copy link
Collaborator Author

stas00 commented Oct 22, 2025

Thank you for reviewing, Masahiro

NirSonnenschein pushed a commit to NirSonnenschein/DeepSpeed that referenced this pull request Oct 26, 2025
Ulysses/ALST integration with HF Accelerate:
- Allow `UlyssesSPAttentionHF.register_with_transformers` to get a
`model` obj as an argument, to match HF accelerate's workflow
- Fix existing Ulysses' tests to tests z2 instead of z1
- Improve documentation
- Add a defensive check

The HF Accelerate PR that depends on this PR is here
huggingface/accelerate#3817

---------

Signed-off-by: Stas Bekman <[email protected]>
Signed-off-by: Nir Sonnenschein <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants