Skip to content

Qwen3-Coder-Next long-context LoRA fine-tuning example - #90

Merged
rcano-baseten merged 2 commits into
mainfrom
rcano/qwen3-coder-next-long-context
Mar 12, 2026
Merged

Qwen3-Coder-Next long-context LoRA fine-tuning example#90
rcano-baseten merged 2 commits into
mainfrom
rcano/qwen3-coder-next-long-context

Conversation

@rcano-baseten

Copy link
Copy Markdown
Collaborator

Summary

  • Adds qwen3-80b-msswift example for fine-tuning Qwen3-Coder-Next (80B MoE) with LoRA using MS-Swift + MegatronLM
  • Default config: 1 node, 8x H200, 32K sequence length, ep=8, LoRA rank 8
  • README includes tested configurations table covering 16K–128K sequence lengths across 1–4 node setups
  • Uses flash-linear-attention, optimizer CPU offload, and activation recomputation for memory efficiency

Tested Configurations

Nodes GPUs Seq Length TP PP EP Peak Memory Status
1 8 16K 8 105 GiB
1 8 32K 8 121 GiB
2 16 64K 2 2 4 106 GiB
4 32 48K 2 16 98 GiB
4 32 64K 2 2 4 118 GiB
4 32 128K 2 4 4 134 GiB

Supersedes #86.

Test plan

  • All configurations above verified end-to-end (training + checkpoint save)
  • Verify truss train init --examples qwen3-80b-msswift works after merge

🤖 Generated with Claude Code

Add qwen3-80b-msswift example for fine-tuning Qwen3-Coder-Next (80B MoE)
with LoRA using MS-Swift + MegatronLM. Default config runs on 1x H200
node at 32K sequence length. Includes tested configurations from 16K
(1 node) to 128K (4 nodes) with detailed parallelism settings.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

@ervinwang-jpg ervinwang-jpg left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm

@rcano-baseten
rcano-baseten merged commit 82ce2a9 into main Mar 12, 2026
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.

2 participants