Skip to content

GitLab

Commit 5c5f67a2 authored by Stefan Schweter's avatar Stefan Schweter
Browse files

modeling: use S3 location for XLM-RoBERTa model

parent db90e121
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
transformers/modeling_xlm_roberta.py
View file @ 5c5f67a2
...@@ -27,8 +27,8 @@ from .file_utils import add_start_docstrings ...@@ -27,8 +27,8 @@ from .file_utils import add_start_docstrings
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP = { XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP = {
'xlm-roberta-base': "https://schweter.eu/cloud/transformers/xlm-roberta-base-pytorch_model.bin", 'xlm-roberta-base': "https://s3.amazonaws.com/models.huggingface.co/bert/xlm-roberta-base-pytorch_model.bin",
'xlm-roberta-large': "https://schweter.eu/cloud/transformers/xlm-roberta-large-pytorch_model.bin", 'xlm-roberta-large': "https://s3.amazonaws.com/models.huggingface.co/bert/xlm-roberta-large-pytorch_model.bin",
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment