Skip to content

Add Qwen3.8-27B VL-text loader and WISE padding/thinking fixes - #715

Open
Zhang-zu-hao wants to merge 3 commits into
zjunlp:mainfrom
Zhang-zu-hao:feat/wise-ft-loss-padding
Open

Add Qwen3.8-27B VL-text loader and WISE padding/thinking fixes#715
Zhang-zu-hao wants to merge 3 commits into
zjunlp:mainfrom
Zhang-zu-hao:feat/wise-ft-loss-padding

Conversation

@Zhang-zu-hao

Copy link
Copy Markdown

Summary

  • Add a Hub-free loader for Qwen3.8-27B (Qwen3_5ForConditionalGeneration, VL shell + 64-layer hybrid LM). This is not pure-text Qwen3.5-9B (model.layers.{}); the edit path is model.language_model.layers.{}.mlp.down_proj.
  • Fix WISE prompt/target location for right padding, wrap enable_thinking=False, and run the edit loop in model.train() so transformers 5.8 gradient checkpointing actually runs.
  • Add hparams/WISE/qwen3.8-27b.yaml (padding_side: right, enable_thinking: false, layer 53). FT/GRACE yamls are structural only and untested. ROME is .example only.

Detect Qwen3_5ForConditionalGeneration via AutoConfig before the qwen2/qwen3 CausalLM branch so Qwen3.5-9B is unchanged, and document WISE/FT/GRACE plus ROME scan scaffolding.
… edit example.

Right-pad label masks no longer count trailing pads; edit-time train() enables transformers 5.8 gradient checkpointing; optional yaml fields keep old configs loading.
Strip local paths, keep Hub-free loader detection, restore WISE train/eval
with try/finally, and record teacher-forcing vs free-generation metrics.
@Zhang-zu-hao

Copy link
Copy Markdown
Author

Hi @pengzju @XeeKee @zxlzr — gentle ping when you have a moment. This PR has been open a couple of days with no review yet; a quick look would be much appreciated. Thanks!

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