Fix TF savedmodel in Roberta (#7795)
* Remove wrong parameter. * Same in Longformer
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user