Update run_glue.py

add DistilBert model shortcut into ALL_MODELS
huggingface · Oct 3, 2019 · 7af0777 · 7af0777
1 parent c1689ac
commit 7af0777
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/run_glue.py b/examples/run_glue.py
@@ -53,7 +53,8 @@
 
 logger = logging.getLogger(__name__)
 
-ALL_MODELS = sum((tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, XLNetConfig, XLMConfig, RobertaConfig)), ())
+ALL_MODELS = sum((tuple(conf.pretrained_config_archive_map.keys()) for conf in (BertConfig, XLNetConfig, XLMConfig, 
+                                                                                RobertaConfig, DistilBertConfig)), ())
 
 MODEL_CLASSES = {
     'bert': (BertConfig, BertForSequenceClassification, BertTokenizer),