Export ALBERT main layer in TensorFlow (#3354)

2020-03-19 13:53:05 -04:00
parent 3bedfd3347
commit f049be7ad4
1 changed files with 1 additions and 0 deletions
--- 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,