Properly propagate `model_inputs`
#14
by Alnusjaponica - opened
This PR fixes mo2ForCausalLM.prepare_inputs_for_generation, which overrides transformers' GenerationMixin.prepare_inputs_for_generation, to ensure that the arguments of Plamo2ForCausalLM.forward are properly propagated.
Alnusjaponica changed pull request status to open
LGTM
yhirokawa changed pull request status to merged