preparing release distil-mBERT

fb0d2f1d · VictorSanh · 5bfcd048 · fb0d2f1d
Commit fb0d2f1d authored Dec 05, 2019 by VictorSanh
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

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

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

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