Skip to content

fix(h3): drop trainer-side LoRA collector; keep PEFT names on IPC - #226

Open
niehen6174 wants to merge 6 commits into
mainfrom
fix/drop-h3-lora-collector
Open

fix(h3): drop trainer-side LoRA collector; keep PEFT names on IPC#226
niehen6174 wants to merge 6 commits into
mainfrom
fix/drop-h3-lora-collector

Conversation

@niehen6174

@niehen6174 niehen6174 commented Sep 1, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Remove h3_weight_key_mapper and the H3 LoRA IPC grouper.
  • Trainer now pushes PEFT/diffusers names (transformer_blocks.*.attn.to_q, ff.net.0.proj, …).
  • sglang-d lora_merge maps those names and applies the disk-load FFN swap.
  • H3 still requires --use-lora --lora-ipc-weight-sync (SFT / --train-only unchanged).

Depends on the matching sglang-d change on sglang-miles-h3.

ci-sglang-pr: #37350

Test plan

  • pytest tests/fast/backends/fsdp_utils/test_peft_lora_key_mapper.py

sglang-d lora_merge now maps PEFT/diffusers names and applies the FFN
swap, so H3 no longer needs a dedicated IPC grouper.
@Rockdu Rockdu added run-ci-e2e Run e2e metric-regression tests on this PR and removed run-ci-e2e Run e2e metric-regression tests on this PR labels Sep 4, 2026

@Rockdu Rockdu left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

LGTM

@Rockdu

Rockdu commented Sep 7, 2026

Copy link
Copy Markdown
Collaborator

will merge after sgl-project/sglang#37350 is merged

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

run-ci-e2e Run e2e metric-regression tests on this PR

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants