Export ALBERT main layer in TensorFlow (#3354)

f049be7a · Lysandre Debut · GitHub · 3bedfd33 · f049be7a
Unverified Commit f049be7a authored Mar 19, 2020 by Lysandre Debut Committed by GitHub Mar 19, 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
@@ -434,6 +434,7 @@ if is_tf_available():

    from .modeling_tf_albert import (
        TFAlbertPreTrainedModel,
+        TFAlbertMainLayer,
        TFAlbertModel,
        TFAlbertForMaskedLM,
        TFAlbertForSequenceClassification,