From fd405e9a93f066cf1230ce4d53e2ade73c4a5497 Mon Sep 17 00:00:00 2001 From: Kevin Canwen Xu Date: Sat, 27 Jun 2020 00:53:10 +0800 Subject: [PATCH] Add BART-base modeling and configuration (#5315) --- src/transformers/configuration_bart.py | 1 + src/transformers/modeling_bart.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/transformers/configuration_bart.py b/src/transformers/configuration_bart.py index 398ff7c4..d07294c7 100644 --- a/src/transformers/configuration_bart.py +++ b/src/transformers/configuration_bart.py @@ -23,6 +23,7 @@ from .configuration_utils import PretrainedConfig logger = logging.getLogger(__name__) BART_PRETRAINED_CONFIG_ARCHIVE_MAP = { + "facebook/bart-base": "https://s3.amazonaws.com/models.huggingface.co/bert/facebook/bart-base/config.json", "facebook/bart-large": "https://s3.amazonaws.com/models.huggingface.co/bert/facebook/bart-large/config.json", "facebook/bart-large-mnli": "https://s3.amazonaws.com/models.huggingface.co/bert/facebook/bart-large-mnli/config.json", "facebook/bart-large-cnn": "https://s3.amazonaws.com/models.huggingface.co/bert/facebook/bart-large-cnn/config.json", diff --git a/src/transformers/modeling_bart.py b/src/transformers/modeling_bart.py index 76f94d01..6f0f8828 100644 --- a/src/transformers/modeling_bart.py +++ b/src/transformers/modeling_bart.py @@ -42,6 +42,7 @@ _TOKENIZER_FOR_DOC = "BartTokenizer" BART_PRETRAINED_MODEL_ARCHIVE_LIST = [ + "facebook/bart-base", "facebook/bart-large", "facebook/bart-large-mnli", "facebook/bart-large-cnn",