templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_{{cookiecutter.lowercase_modelname}}.py/set_output_embeddings
    def set_output_embeddings(self, new_embeddings):
        self.lm_head = new_embeddings