templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_flax_{{cookiecutter.lowercase_modelname}}.py/_get_decoder_module
    def _get_decoder_module(self):
        return self.model.decoder