Config to Model mapping

2020-01-13 20:05:20 +00:00
parent cf8a70bf68
commit b803b067bf
17 changed files with 58 additions and 55 deletions
--- a/examples/summarization/configuration_bertabs.py
+++ b/examples/summarization/configuration_bertabs.py
@@ -62,6 +62,7 @@ class BertAbsConfig(PretrainedConfig):
    """

    pretrained_config_archive_map = BERTABS_FINETUNED_CONFIG_MAP
+    model_type = "bertabs"

    def __init__(
        self,