Refactor Logits Naming
#15
by codecho - opened
This PR adjusts internal logits naming in MoonshotKimiaForCausalLM: lm_head output is now text_logits, and mimo_output output is audio_logits. This modification does not change the effective output (as variable names and their corresponding order in the returned tuple were adjusted synchronously), aiming to resolve naming confusion encountered during further development. The usage in the inference script kimia_infer/api/kimia.py at line 114 – audio_logits, text_logits, ... = self.alm.forward(...), which assigns the first returned value to audio_logits (for audio) and the second to text_logits (for text) – also validates the rationale for this change.
LGTM
bigmoyan changed pull request status to merged