From f049be7ad49e3d5d2bccddb675f58c1562d60ff3 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Thu, 19 Mar 2020 13:53:05 -0400 Subject: [PATCH] Export ALBERT main layer in TensorFlow (#3354) --- src/transformers/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index a41ae430dc..e2332654e3 100755 --- 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,