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