module: import CamembertForTokenClassification

2019-11-18 14:14:54 +01:00
parent d32ce2c8df
commit 33753d9139
1 changed files with 1 additions and 0 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -100,6 +100,7 @@ if is_torch_available():
                                DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_camembert import (CamembertForMaskedLM, CamembertModel,
                                CamembertForSequenceClassification, CamembertForMultipleChoice,
+                                CamembertForTokenClassification,
                                CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_encoder_decoder import PreTrainedEncoderDecoder, Model2Model