Skip to content

GitLab

Unverified Commit f049be7a authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Export ALBERT main layer in TensorFlow (#3354)

parent 3bedfd33
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/__init__.py
View file @ f049be7a
...@@ -434,6 +434,7 @@ if is_tf_available(): ...@@ -434,6 +434,7 @@ if is_tf_available():
from .modeling_tf_albert import ( from .modeling_tf_albert import (
TFAlbertPreTrainedModel, TFAlbertPreTrainedModel,
TFAlbertMainLayer,
TFAlbertModel, TFAlbertModel,
TFAlbertForMaskedLM, TFAlbertForMaskedLM,
TFAlbertForSequenceClassification, TFAlbertForSequenceClassification,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment