From eb59e9f70513b538d2174d4ea1efea7ba8554b58 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Wed, 15 Jan 2020 16:28:50 +0000 Subject: [PATCH] Graduate sst-2 to a canonical one --- src/transformers/configuration_distilbert.py | 1 + src/transformers/modeling_distilbert.py | 1 + src/transformers/modeling_tf_distilbert.py | 1 + src/transformers/pipelines.py | 6 +++--- tests/test_pipelines.py | 8 ++++---- 5 files changed, 10 insertions(+), 7 deletions(-) diff --git a/src/transformers/configuration_distilbert.py b/src/transformers/configuration_distilbert.py index 239e2de08f..11f14a85c9 100644 --- a/src/transformers/configuration_distilbert.py +++ b/src/transformers/configuration_distilbert.py @@ -27,6 +27,7 @@ DISTILBERT_PRETRAINED_CONFIG_ARCHIVE_MAP = { "distilbert-base-uncased-distilled-squad": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-distilled-squad-config.json", "distilbert-base-german-cased": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-german-cased-config.json", "distilbert-base-multilingual-cased": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-multilingual-cased-config.json", + "distilbert-base-uncased-finetuned-sst-2-english": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-config.json", } diff --git a/src/transformers/modeling_distilbert.py b/src/transformers/modeling_distilbert.py index 475992d382..b084103f9f 100644 --- a/src/transformers/modeling_distilbert.py +++ b/src/transformers/modeling_distilbert.py @@ -40,6 +40,7 @@ DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP = { "distilbert-base-uncased-distilled-squad": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-distilled-squad-pytorch_model.bin", "distilbert-base-german-cased": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-german-cased-pytorch_model.bin", "distilbert-base-multilingual-cased": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-multilingual-cased-pytorch_model.bin", + "distilbert-base-uncased-finetuned-sst-2-english": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-pytorch_model.bin", } diff --git a/src/transformers/modeling_tf_distilbert.py b/src/transformers/modeling_tf_distilbert.py index f588a9b2af..3118411d89 100644 --- a/src/transformers/modeling_tf_distilbert.py +++ b/src/transformers/modeling_tf_distilbert.py @@ -34,6 +34,7 @@ TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP = { "distilbert-base-uncased": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-tf_model.h5", "distilbert-base-uncased-distilled-squad": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-distilled-squad-tf_model.h5", "distilbert-base-multilingual-cased": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-multilingual-cased-tf_model.h5", + "distilbert-base-uncased-finetuned-sst-2-english": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-tf_model.h5", } diff --git a/src/transformers/pipelines.py b/src/transformers/pipelines.py index 3049ff85e9..4eb5d5d1a6 100755 --- a/src/transformers/pipelines.py +++ b/src/transformers/pipelines.py @@ -870,10 +870,10 @@ SUPPORTED_TASKS = { "pt": AutoModelForSequenceClassification if is_torch_available() else None, "default": { "model": { - "pt": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-pytorch_model.bin", - "tf": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-tf_model.h5", + "pt": "distilbert-base-uncased-finetuned-sst-2-english", + "tf": "distilbert-base-uncased-finetuned-sst-2-english", }, - "config": "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-config.json", + "config": "distilbert-base-uncased-finetuned-sst-2-english", "tokenizer": "distilbert-base-uncased", }, }, diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py index 936fbd8813..717744bc94 100644 --- a/tests/test_pipelines.py +++ b/tests/test_pipelines.py @@ -49,16 +49,16 @@ TF_FEATURE_EXTRACT_FINETUNED_MODELS = { TF_TEXT_CLASSIF_FINETUNED_MODELS = { ( "bert-base-uncased", - "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-tf_model.h5", - "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-config.json", + "distilbert-base-uncased-finetuned-sst-2-english", + "distilbert-base-uncased-finetuned-sst-2-english", ) } TEXT_CLASSIF_FINETUNED_MODELS = { ( "bert-base-uncased", - "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-pytorch_model.bin", - "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-finetuned-sst-2-english-config.json", + "distilbert-base-uncased-finetuned-sst-2-english", + "distilbert-base-uncased-finetuned-sst-2-english", ) }