Fix e_score_correction_bias wrong dtype

#33
by fxmarty-amd - opened

As per title.

In the checkpoint, mlp.gate.e_score_correction_bias is in float32.

Simply using torch.empty without specifying the dtype risks initializing this parameter with a wrong dtype.

This is similar to https://huggingface.co/moonshotai/Kimi-K2.5/discussions/105.

cc @bigmoyan could you check these PRs? Thanks a lot

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment