distilbert: add pytorch model for new German distilbert model

22333945 · Stefan Schweter · 33780278 · 22333945
Commit 22333945 authored Nov 19, 2019 by Stefan Schweter
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

transformers/modeling_distilbert.py transformers/modeling_distilbert.py +2 -1

No files found.
--- a/transformers/modeling_distilbert.py
+++ b/transformers/modeling_distilbert.py
@@ -42,7 +42,8 @@ logger = logging.getLogger(__name__)

 DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP = {
    'distilbert-base-uncased': "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-pytorch_model.bin",
-    'distilbert-base-uncased-distilled-squad': "https://s3.amazonaws.com/models.huggingface.co/bert/distilbert-base-uncased-distilled-squad-pytorch_model.bin"
+    '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"
 }