Fix TF savedmodel in Roberta (#7795)

* Remove wrong parameter.

* Same in Longformer
This commit is contained in:
Julien Plu
2020-10-14 23:48:50 +02:00
committed by GitHub
parent 3032de9369
commit 3a134f7c67
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -125,7 +125,7 @@ class TFLongformerEmbeddings(tf.keras.layers.Layer):
"""
def __init__(self, config, **kwargs):
super().__init__(config, **kwargs)
super().__init__(**kwargs)
self.padding_idx = 1
self.vocab_size = config.vocab_size
+1 -1
View File
@@ -70,7 +70,7 @@ class TFRobertaEmbeddings(tf.keras.layers.Layer):
"""
def __init__(self, config, **kwargs):
super().__init__(config, **kwargs)
super().__init__(**kwargs)
self.padding_idx = 1
self.vocab_size = config.vocab_size