From 9972562d33dd6455fddc41d1f164c5b84801a46a Mon Sep 17 00:00:00 2001
From: kumapo <kumapo@users.noreply.github.com>
Date: Thu, 7 May 2020 01:00:23 +0900
Subject: [PATCH] Include ElectraPreTrainedModel into __init__ (#4173)

---
 src/transformers/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
index b5270b96f0..d74e55e106 100755
--- 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,