Export the public API surface (#64)

keras-team · May 7, 2024 · e29a3bf · e29a3bf
1 parent 4d14263
commit e29a3bf
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/keras_nlp/api/models/__init__.py b/keras_nlp/api/models/__init__.py
@@ -152,6 +152,16 @@
 )
 from keras_nlp.src.models.opt.opt_preprocessor import OPTPreprocessor
 from keras_nlp.src.models.opt.opt_tokenizer import OPTTokenizer
+from keras_nlp.src.models.paligemma.pali_gemma_backbone import PaliGemmaBackbone
+from keras_nlp.src.models.paligemma.pali_gemma_causal_lm import (
+    PaliGemmaCausalLM,
+)
+from keras_nlp.src.models.paligemma.pali_gemma_causal_lm_preprocesor import (
+    PaliGemmaCausalLMPreprocessor,
+)
+from keras_nlp.src.models.paligemma.pali_gemma_tokenizer import (
+    PaliGemmaTokenizer,
+)
 from keras_nlp.src.models.preprocessor import Preprocessor
 from keras_nlp.src.models.roberta.roberta_backbone import RobertaBackbone
 from keras_nlp.src.models.roberta.roberta_classifier import RobertaClassifier