Skip to content

[Regression] Qwen3.5 saved checkpoint is not correct with save_pretrained API since version 5.4.0 #45216

Closed

Description

@xin3he

System Info

transformers == 5.3.0 works well
transformers ==5.4.0 returns Unexpected model.language_model.language_model.language_model.layers.7.self_attn.v_proj.weight in loaded safetensors file

Who can help?

@zucchini-nlp

Information

  • The official example scripts
  • My own modified scripts

Tasks

  • An officially supported task in the examples folder (such as GLUE/SQuAD, ...)
  • My own task or dataset (give details below)

Reproduction

import transformers

name = "Qwen/Qwen3.5-0.8B"
model = transformers.Qwen3_5ForConditionalGeneration.from_pretrained(name, trust_remote_code=True)
model.save_pretrained("./qwen-35")

from safetensors.torch import save_file, load_file

loaded = load_file("./qwen-35/model.safetensors")
assert not 'model.language_model.language_model.language_model.layers.7.self_attn.v_proj.weight' in loaded, "Unexpected model.language_model.language_model.language_model.layers.7.self_attn.v_proj.weight in loaded safetensors file"

Expected behavior

assert pass

Activity

  1. added a commit that references this issue on Apr 5, 2026
    9c3053c
  2. zucchini-nlp commented on Apr 7, 2026

    @zucchini-nlp
    Member

    Oh no, some conversions are meant to be applied only for certain classes. A quick workaround could be to bring back the cls attribute for conversion_mapping though as I mentioned earlier to @Cyrilvallez imo we need a better way to store conversions. We have several regex patterns that are model-class specific, for ex applied only when a prediction is present

    Ig in long term we have to adjust conversion dict's layout slightly

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions