From cdd86592317e7db3bab75555c3837fabc74e3429 Mon Sep 17 00:00:00 2001 From: Julien Plu Date: Mon, 8 Feb 2021 14:10:50 +0100 Subject: [PATCH] Fix TF template (#10069) * Fix template * Fix template --- .../modeling_tf_{{cookiecutter.lowercase_modelname}}.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py b/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py index feffe7a585..7ff9a10eb1 100644 --- a/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py +++ b/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/modeling_tf_{{cookiecutter.lowercase_modelname}}.py @@ -305,7 +305,7 @@ class TF{{cookiecutter.camelcase_modelname}}Intermediate(tf.keras.layers.Layer): # Copied from transformers.models.bert.modeling_tf_bert.TFBertOutput with Bert->{{cookiecutter.camelcase_modelname}} class TF{{cookiecutter.camelcase_modelname}}Output(tf.keras.layers.Layer): - def __init__(self, config: BertConfig, **kwargs): + def __init__(self, config: {{cookiecutter.camelcase_modelname}}Config, **kwargs): super().__init__(**kwargs) self.dense = tf.keras.layers.Dense(