Skip to content

Fix checkpoint loading with load_main_params_from_ckpt=True for grouped weight - #4324

Merged
yaox12 merged 6 commits into
NVIDIA:devfrom
ksivaman:fix_checkpointing_with_disopt_loader
May 7, 2026
Merged

Fix checkpoint loading with load_main_params_from_ckpt=True for grouped weight#4324
yaox12 merged 6 commits into
NVIDIA:devfrom
ksivaman:fix_checkpointing_with_disopt_loader

Commits

Commits on Apr 14, 2026

Commits on Apr 16, 2026

Commits on Apr 28, 2026

Commits on May 6, 2026

Commits on May 7, 2026

  • fix lint

    committed