Skip to content

GitLab

Unverified Commit 1e3c3622 authored by Julien Plu's avatar Julien Plu Committed by GitHub
Browse files

Fix template (#9512)

parent d415882b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py
View file @ 1e3c3622
......@@ -462,7 +462,7 @@ class TF{{cookiecutter.camelcase_modelname}}LMPredictionHead(tf.keras.layers.Lay
super().build(input_shape)
def get_output_embeddings(self):
return self.input_embeddings.word_embeddings
return self.input_embeddings
def set_output_embeddings(self, value):
self.input_embeddings.word_embeddings = value
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment