update

modeling_bailing_moe.py

@@ -1444,6 +1444,7 @@ class BailingMoeForCausalLM(BailingMoePreTrainedModel):
             logits = F.linear(hidden_states, norm_weight, None)
         else:
             logits = self.lm_head(hidden_states)
+        return logits
 
     @add_start_docstrings_to_model_forward(BAILINGMOE_INPUTS_DOCSTRING)
     @replace_return_docstrings(output_type=MoeCausalLMOutputWithPast, config_class=_CONFIG_FOR_DOC)