Skip to content

v4.5.3

Latest

Choose a tag to compare

@hjh0119 hjh0119 released this 08 Sep 13:00
· 7 commits to main since this release

New Models

New Features

  • Megatron-SWIFT Muon optimizer supports different learning rates for different parameters.
  • OPD distillation (GRPO) now supports separate vision inputs for teacher and student, enabling multimodal distillation. Reference doc: https://swift.readthedocs.io/en/latest/Instruction/Distillation.html
  • A last-checkpoint link is maintained when saving checkpoints, making resume easier.
  • dataloader_persistent_workers now defaults to True to avoid dataloader cold start on each eval.
  • Performance: vectorized padding-free log-prob restore and sequence reductions; Megatron loss all-reduce deferred to logging time; vectorized ring attention LSE extraction on NPU.
  • Added the vllm_kunlun inference backend for rollout/deploy.
  • Removed the gradio dependency.

Bug Fixes

  • Training: resume_from_checkpoint with transformers 4.x, DPO/IPO log-prob normalization, reward model margins aligned with pairs, predict_with_generate DDP evaluation, loss scaling with sequence parallelism, vocab loss & external files in multi-processing, template pickling, py3.14 compatibility, etc.
  • RL: Megatron GRPO vocab-parallel log-prob gradients, vLLM weight sync in colocate mode (LoRA IndexError, sender device alignment), colocated vLLM cleanup ordering on rollout failures, token-in/token-out consistency in agentic multi-step rollouts.
  • Megatron: router replay CP partition and expert indices, MTP shared weights forward, reference adapter loading, zero multimodal learning rates, Megatron-FSDP DTensor checkpoint save/load, etc.
  • Model fixes: Qwen3.5 (linear attention FP32 weights, multimodal packing kwargs), Qwen-Omni video fps, MiniCPM-o 4.5 video inputs, mplug_owl2_1, etc.

What's Changed

New Contributors

Full Changelog: v4.5.2...v4.5.3