Include ElectraPreTrainedModel into __init__ (#4173)

2020-05-07 01:00:23 +09:00
parent ff8ed52dd8
commit 9972562d33
1 changed files with 1 additions and 0 deletions
--- a/src/transformers/init.py
+++ b/src/transformers/init.py
@@ -312,6 +312,7 @@ if is_torch_available():
        ElectraForPreTraining,
        ElectraForMaskedLM,
        ElectraForTokenClassification,
+        ElectraPreTrainedModel,
        ElectraModel,
        load_tf_weights_in_electra,
        ELECTRA_PRETRAINED_MODEL_ARCHIVE_MAP,