Skip to content

Commit 403ffd2

Browse files
authored
Add missing exports to models API (#763)
We are re-exporting deberta with a note about TPU performance We are exporting albert and fnet classifiers for the first time.
1 parent 59fe789 commit 403ffd2

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

keras_nlp/models/__init__.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
# limitations under the License.
1414

1515
from keras_nlp.models.albert.albert_backbone import AlbertBackbone
16+
from keras_nlp.models.albert.albert_classifier import AlbertClassifier
1617
from keras_nlp.models.albert.albert_masked_lm import AlbertMaskedLM
1718
from keras_nlp.models.albert.albert_masked_lm_preprocessor import (
1819
AlbertMaskedLMPreprocessor,
@@ -25,6 +26,14 @@
2526
from keras_nlp.models.bert.bert_classifier import BertClassifier
2627
from keras_nlp.models.bert.bert_preprocessor import BertPreprocessor
2728
from keras_nlp.models.bert.bert_tokenizer import BertTokenizer
29+
from keras_nlp.models.deberta_v3.deberta_v3_backbone import DebertaV3Backbone
30+
from keras_nlp.models.deberta_v3.deberta_v3_classifier import (
31+
DebertaV3Classifier,
32+
)
33+
from keras_nlp.models.deberta_v3.deberta_v3_preprocessor import (
34+
DebertaV3Preprocessor,
35+
)
36+
from keras_nlp.models.deberta_v3.deberta_v3_tokenizer import DebertaV3Tokenizer
2837
from keras_nlp.models.distil_bert.distil_bert_backbone import DistilBertBackbone
2938
from keras_nlp.models.distil_bert.distil_bert_classifier import (
3039
DistilBertClassifier,
@@ -36,6 +45,7 @@
3645
DistilBertTokenizer,
3746
)
3847
from keras_nlp.models.f_net.f_net_backbone import FNetBackbone
48+
from keras_nlp.models.f_net.f_net_classifier import FNetClassifier
3949
from keras_nlp.models.f_net.f_net_preprocessor import FNetPreprocessor
4050
from keras_nlp.models.f_net.f_net_tokenizer import FNetTokenizer
4151
from keras_nlp.models.gpt2.gpt2_backbone import GPT2Backbone

0 commit comments

Comments
 (0)