Skip to content

GitLab

Commit 1806eabf authored by Stefan Schweter's avatar Stefan Schweter
Browse files

module: add DistilBertForTokenClassification import

parent 1c7253cc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
transformers/__init__.py
View file @ 1806eabf
...@@ -93,6 +93,7 @@ if is_torch_available(): ...@@ -93,6 +93,7 @@ if is_torch_available():
ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP) ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP)
from .modeling_distilbert import (DistilBertForMaskedLM, DistilBertModel, from .modeling_distilbert import (DistilBertForMaskedLM, DistilBertModel,
DistilBertForSequenceClassification, DistilBertForQuestionAnswering, DistilBertForSequenceClassification, DistilBertForQuestionAnswering,
DistilBertForTokenClassification,
DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP) DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
from .modeling_encoder_decoder import PreTrainedEncoderDecoder, Model2Model from .modeling_encoder_decoder import PreTrainedEncoderDecoder, Model2Model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment