From b1116fd6736bbd69bd10aacb1de3a0e98985fb2d Mon Sep 17 00:00:00 2001 From: Gunnlaugur Thor Briem Date: Tue, 3 Mar 2020 23:05:40 +0000 Subject: [PATCH] fix: passing config as Layer trainable param Lurking bugs discovered while working on other stuff. --- src/transformers/modeling_tf_albert.py | 2 +- src/transformers/modeling_tf_openai.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/modeling_tf_albert.py b/src/transformers/modeling_tf_albert.py index 64c9dad06a..75f785ede0 100644 --- a/src/transformers/modeling_tf_albert.py +++ b/src/transformers/modeling_tf_albert.py @@ -480,7 +480,7 @@ class TFAlbertMLMHead(tf.keras.layers.Layer): class TFAlbertMainLayer(tf.keras.layers.Layer): def __init__(self, config, **kwargs): - super().__init__(config, **kwargs) + super().__init__(**kwargs) self.num_hidden_layers = config.num_hidden_layers self.embeddings = TFAlbertEmbeddings(config, name="embeddings") diff --git a/src/transformers/modeling_tf_openai.py b/src/transformers/modeling_tf_openai.py index f04104db83..6a97ae7786 100644 --- a/src/transformers/modeling_tf_openai.py +++ b/src/transformers/modeling_tf_openai.py @@ -199,7 +199,7 @@ class TFBlock(tf.keras.layers.Layer): class TFOpenAIGPTMainLayer(tf.keras.layers.Layer): def __init__(self, config, *inputs, **kwargs): - super().__init__(config, *inputs, **kwargs) + super().__init__(*inputs, **kwargs) self.output_hidden_states = config.output_hidden_states self.output_attentions = config.output_attentions self.num_hidden_layers = config.n_layer