templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py/set_embed_tokens
    def set_embed_tokens(self, embed_tokens):
        self.embed_tokens = embed_tokens