From f1822bb14ffc87db4dfa6e445580826e102f8e89 Mon Sep 17 00:00:00 2001 From: Matt Watson Date: Wed, 6 Dec 2023 08:35:04 +0000 Subject: [PATCH] Fix opt get_config to chain to super --- keras_nlp/models/opt/opt_backbone.py | 22 +++++++++++++--------- 1 file changed, 13 insertions(+), 9 deletions(-) diff --git a/keras_nlp/models/opt/opt_backbone.py b/keras_nlp/models/opt/opt_backbone.py index ff1495ba9f..27f9543d30 100644 --- a/keras_nlp/models/opt/opt_backbone.py +++ b/keras_nlp/models/opt/opt_backbone.py @@ -151,15 +151,19 @@ def __init__( self.token_embedding = embedding_layer.token_embedding def get_config(self): - return { - "vocabulary_size": self.vocabulary_size, - "num_layers": self.num_layers, - "num_heads": self.num_heads, - "hidden_dim": self.hidden_dim, - "intermediate_dim": self.intermediate_dim, - "dropout": self.dropout, - "max_sequence_length": self.max_sequence_length, - } + config = super().get_config() + config.update( + { + "vocabulary_size": self.vocabulary_size, + "num_layers": self.num_layers, + "num_heads": self.num_heads, + "hidden_dim": self.hidden_dim, + "intermediate_dim": self.intermediate_dim, + "dropout": self.dropout, + "max_sequence_length": self.max_sequence_length, + } + ) + return config @classproperty def presets(cls):