Include ElectraPreTrainedModel into __init__ (#4173)

Include ElectraPreTrainedModel into init (#4173)
9972562d · kumapo · GitHub · ff8ed52d · 9972562d
Unverified Commit 9972562d authored May 07, 2020 by kumapo Committed by GitHub May 06, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/__init__.py src/transformers/__init__.py +1 -0

No files found.
--- 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,