Add BART-base modeling and configuration (#5315)

fd405e9a · Kevin Canwen Xu · GitHub · 798dbff6 · fd405e9a · fd405e9a
Unverified Commit fd405e9a authored Jun 27, 2020 by Kevin Canwen Xu Committed by GitHub Jun 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/transformers/configuration_bart.py src/transformers/configuration_bart.py +1 -0

src/transformers/modeling_bart.py src/transformers/modeling_bart.py +1 -0

No files found.
--- 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",

--- 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",