c-my/transformers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add BART-base modeling and configuration (#5315)

Browse Source
This commit is contained in:
Kevin Canwen Xu
2020-06-27 00:53:10 +08:00
committed by GitHub
parent 798dbff6a7
commit fd405e9a93
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/transformers/configuration_bart.py
+1
View File
@@ -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",
src/transformers/modeling_bart.py
+1
View File
@@ -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",