JuncaiL's picture
fix state_dict loading in MoE model
d8d97b0 verified