SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Include ElectraPreTrainedModel into __init__ (#4173)

Browse Source
This commit is contained in:
kumapo
2020-05-07 01:00:23 +09:00
committed by GitHub
parent ff8ed52dd8
commit 9972562d33
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/__init__.py
View File

@@ -312,6 +312,7 @@ if is_torch_available():
ElectraForPreTraining, ElectraForPreTraining,
ElectraForMaskedLM, ElectraForMaskedLM,
ElectraForTokenClassification, ElectraForTokenClassification,
ElectraPreTrainedModel,
ElectraModel, ElectraModel,
load_tf_weights_in_electra, load_tf_weights_in_electra,
ELECTRA_PRETRAINED_MODEL_ARCHIVE_MAP, ELECTRA_PRETRAINED_MODEL_ARCHIVE_MAP,