Upload config.py with huggingface_hub

config.py

@@ -225,14 +225,25 @@ class HATArchitectureConfig(PretrainedConfig):
         decoder_config = DecoderHATModelConfig.from_dict(decoder_dict) if decoder_dict else None
         special_token_dict = config_dict.pop("special_token_dict", {"<|eot_id|>": 192})
         max_word_size = config_dict.pop("max_word_size", 100)
-        return cls(
-            encoder_config=encoder_config,
-            backbone_config=backbone_config,
-            decoder_config=decoder_config,
-            special_token_dict=special_token_dict,
-            max_word_size=max_word_size,
-            **config_dict,
-        ), {}
+        return_unused_kwargs = config_dict.pop("return_unused_kwargs", False)
+        if return_unused_kwargs:
+            return cls(
+                encoder_config=encoder_config,
+                backbone_config=backbone_config,
+                decoder_config=decoder_config,
+                special_token_dict=special_token_dict,
+                max_word_size=max_word_size,
+                **config_dict,
+            ), {}
+        else:
+            return cls(
+                encoder_config=encoder_config,
+                backbone_config=backbone_config,
+                decoder_config=decoder_config,
+                special_token_dict=special_token_dict,
+                max_word_size=max_word_size,
+                **config_dict,
+            )
 
     def to_dict(self):
         config_dict = {}