Skip to content

Update SpeechLM2 for Automodel main - #16021

Open
pzelasko wants to merge 5 commits into
mainfrom
codex/update-automodel-main
Open

Update SpeechLM2 for Automodel main#16021
pzelasko wants to merge 5 commits into
mainfrom
codex/update-automodel-main

Conversation

@pzelasko

@pzelasko pzelasko commented Aug 4, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Update SpeechLM2 to Automodel main at ee0b2261d80d395df7f2dc4767f529cc6b757254 and Megatron-FSDP 0.5.0.
  • Migrate distributed loading and model setup to Automodel's public DistributedSetup API.
  • Remove obsolete Nemotron-V3 compatibility code and the inline shared-experts patch.
  • Raise compatible dependency floors and refresh Docker, configs, docs, tests, and uv.lock.

Validation

  • Focused migration tests: 59 passed, 2 skipped.
  • isort and Black 24.10.0 checks pass.
  • Python syntax compilation and git diff checks pass.
  • uv lock --check passes.

@copy-pr-bot

copy-pr-bot Bot commented Aug 4, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@pzelasko
pzelasko marked this pull request as ready for review August 7, 2026 21:21
@pzelasko
pzelasko requested a review from a team as a code owner August 7, 2026 21:21
@pzelasko

pzelasko commented Aug 7, 2026

Copy link
Copy Markdown
Collaborator Author

/ok to test 3e29e99

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