module: add DistilBertForTokenClassification import

2019-11-11 16:18:48 +01:00
parent 1c7253cc5f
commit 1806eabf59
1 changed files with 1 additions and 0 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -93,6 +93,7 @@ if is_torch_available():
                                ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_distilbert import (DistilBertForMaskedLM, DistilBertModel,
                                DistilBertForSequenceClassification, DistilBertForQuestionAnswering,
+                                DistilBertForTokenClassification,
                                DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_encoder_decoder import PreTrainedEncoderDecoder, Model2Model