keras-hub-nightly 0.15.0.dev20240911134614__tar.gz → 0.16.0.dev2024092017__tar.gz

keras_hub_nightly-0.16.0.dev2024092017/PKG-INFO

@@ -0,0 +1,202 @@
+Metadata-Version: 2.1
+Name: keras_hub-nightly
+Version: 0.16.0.dev2024092017
+Summary: Industry-strength Natural Language Processing extensions for Keras.
+Home-page: https://github.com/keras-team/keras-hub
+Author: Keras team
+Author-email: keras-hub@google.com
+License: Apache License 2.0
+Classifier: Development Status :: 3 - Alpha
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Operating System :: Unix
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: MacOS
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Software Development
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: absl-py
+Requires-Dist: numpy
+Requires-Dist: packaging
+Requires-Dist: regex
+Requires-Dist: rich
+Requires-Dist: kagglehub
+Requires-Dist: tensorflow-text; platform_system != "Darwin"
+Provides-Extra: extras
+Requires-Dist: rouge-score; extra == "extras"
+Requires-Dist: sentencepiece; extra == "extras"
+
+# KerasNLP: Multi-framework NLP Models
+[![](https://github.com/keras-team/keras-hub/workflows/Tests/badge.svg?branch=master)](https://github.com/keras-team/keras-hub/actions?query=workflow%3ATests+branch%3Amaster)
+![Python](https://img.shields.io/badge/python-v3.9.0+-success.svg)
+[![contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat)](https://github.com/keras-team/keras-hub/issues)
+
+> [!IMPORTANT]
+> 📢 KerasNLP is becoming KerasHub! 📢 Read
+> [the announcement](https://github.com/keras-team/keras-hub/issues/1831).
+>
+> We have renamed the repo to KerasHub in preparation for the release, but have not yet
+> released the new package. Follow the announcement for news.
+
+KerasNLP is a natural language processing library that works natively
+with TensorFlow, JAX, or PyTorch. KerasNLP provides a repository of pre-trained
+models and a collection of lower-level building blocks for language modeling.
+Built on Keras 3, models can be trained and serialized in any framework
+and re-used in another without costly migrations.
+
+This library is an extension of the core Keras API; all high-level modules are
+Layers and Models that receive that same level of polish as core Keras.
+If you are familiar with Keras, congratulations! You already understand most of
+KerasNLP.
+
+All models support JAX, TensorFlow, and PyTorch from a single model
+definition and can be fine-tuned on GPUs and TPUs out of the box. Models can
+be trained on individual accelerators with built-in PEFT techniques, or
+fine-tuned at scale with model and data parallel training. See our
+[Getting Started guide](https://keras.io/guides/keras_nlp/getting_started)
+to start learning our API. Browse our models on
+[Kaggle](https://www.kaggle.com/organizations/keras/models).
+We welcome contributions.
+
+## Quick Links
+
+### For everyone
+
+- [Home Page](https://keras.io/keras_nlp)
+- [Developer Guides](https://keras.io/guides/keras_nlp)
+- [API Reference](https://keras.io/api/keras_nlp)
+- [Pre-trained Models](https://www.kaggle.com/organizations/keras/models)
+
+### For contributors
+
+- [Contributing Guide](CONTRIBUTING.md)
+- [Roadmap](ROADMAP.md)
+- [Style Guide](STYLE_GUIDE.md)
+- [API Design Guide](API_DESIGN_GUIDE.md)
+- [Call for Contributions](https://github.com/keras-team/keras-hub/issues?q=is%3Aissue+is%3Aopen+label%3A%22contributions+welcome%22)
+
+## Quickstart
+
+Fine-tune BERT on IMDb movie reviews:
+
+```python
+import os
+os.environ["KERAS_BACKEND"] = "jax"  # Or "tensorflow" or "torch"!
+
+import keras_nlp
+import tensorflow_datasets as tfds
+
+imdb_train, imdb_test = tfds.load(
+    "imdb_reviews",
+    split=["train", "test"],
+    as_supervised=True,
+    batch_size=16,
+)
+# Load a BERT model.
+classifier = keras_nlp.models.Classifier.from_preset(
+    "bert_base_en", 
+    num_classes=2,
+    activation="softmax",
+)
+# Fine-tune on IMDb movie reviews.
+classifier.fit(imdb_train, validation_data=imdb_test)
+# Predict two new examples.
+classifier.predict(["What an amazing movie!", "A total waste of my time."])
+```
+
+Try it out [in a colab](https://colab.research.google.com/gist/mattdangerw/e457e42d5ea827110c8d5cb4eb9d9a07/kerasnlp-quickstart.ipynb).
+For more in depth guides and examples, visit
+[keras.io/keras_nlp](https://keras.io/keras_nlp/).
+
+## Installation
+
+To install the latest KerasNLP release with Keras 3, simply run:
+
+```
+pip install --upgrade keras-nlp
+```
+
+To install the latest nightly changes for both KerasNLP and Keras, you can use
+our nightly package.
+
+```
+pip install --upgrade keras-nlp-nightly
+```
+
+Note that currently, installing KerasNLP will always pull in TensorFlow for use
+of the `tf.data` API for preprocessing. Even when pre-processing with `tf.data`,
+training can still happen on any backend.
+
+Read [Getting started with Keras](https://keras.io/getting_started/) for more
+information on installing Keras 3 and compatibility with different frameworks.
+
+> [!IMPORTANT]
+> We recommend using KerasNLP with TensorFlow 2.16 or later, as TF 2.16 packages
+> Keras 3 by default.
+
+## Configuring your backend
+
+If you have Keras 3 installed in your environment (see installation above),
+you can use KerasNLP with any of JAX, TensorFlow and PyTorch. To do so, set the
+`KERAS_BACKEND` environment variable. For example:
+
+```shell
+export KERAS_BACKEND=jax
+```
+
+Or in Colab, with:
+
+```python
+import os
+os.environ["KERAS_BACKEND"] = "jax"
+
+import keras_nlp
+```
+
+> [!IMPORTANT]
+> Make sure to set the `KERAS_BACKEND` before import any Keras libraries, it
+> will be used to set up Keras when it is first imported.
+
+## Compatibility
+
+We follow [Semantic Versioning](https://semver.org/), and plan to
+provide backwards compatibility guarantees both for code and saved models built
+with our components. While we continue with pre-release `0.y.z` development, we
+may break compatibility at any time and APIs should not be consider stable.
+
+## Disclaimer
+
+KerasNLP provides access to pre-trained models via the `keras_nlp.models` API.
+These pre-trained models are provided on an "as is" basis, without warranties
+or conditions of any kind. The following underlying models are provided by third
+parties, and subject to separate licenses:
+BART, BLOOM, DeBERTa, DistilBERT, GPT-2, Llama, Mistral, OPT, RoBERTa, Whisper,
+and XLM-RoBERTa.
+
+## Citing KerasNLP
+
+If KerasNLP helps your research, we appreciate your citations.
+Here is the BibTeX entry:
+
+```bibtex
+@misc{kerasnlp2022,
+  title={KerasNLP},
+  author={Watson, Matthew, and Qian, Chen, and Bischof, Jonathan and Chollet, 
+  Fran\c{c}ois and others},
+  year={2022},
+  howpublished={\url{https://github.com/keras-team/keras-hub}},
+}
+```
+
+## Acknowledgements
+
+Thank you to all of our wonderful contributors!
+
+<a href="https://github.com/keras-team/keras-hub/graphs/contributors">
+  <img src="https://contrib.rocks/image?repo=keras-team/keras-hub" />
+</a>

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/README.md

@@ -1,10 +1,17 @@
-# KerasHub: Multi-framework NLP Models
+# KerasNLP: Multi-framework NLP Models
 [![](https://github.com/keras-team/keras-hub/workflows/Tests/badge.svg?branch=master)](https://github.com/keras-team/keras-hub/actions?query=workflow%3ATests+branch%3Amaster)
 ![Python](https://img.shields.io/badge/python-v3.9.0+-success.svg)
 [![contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat)](https://github.com/keras-team/keras-hub/issues)
 
-KerasHub is a natural language processing library that works natively
-with TensorFlow, JAX, or PyTorch. KerasHub provides a repository of pre-trained
+> [!IMPORTANT]
+> 📢 KerasNLP is becoming KerasHub! 📢 Read
+> [the announcement](https://github.com/keras-team/keras-hub/issues/1831).
+>
+> We have renamed the repo to KerasHub in preparation for the release, but have not yet
+> released the new package. Follow the announcement for news.
+
+KerasNLP is a natural language processing library that works natively
+with TensorFlow, JAX, or PyTorch. KerasNLP provides a repository of pre-trained
 models and a collection of lower-level building blocks for language modeling.
 Built on Keras 3, models can be trained and serialized in any framework
 and re-used in another without costly migrations.
@@ -12,13 +19,13 @@ and re-used in another without costly migrations.
 This library is an extension of the core Keras API; all high-level modules are
 Layers and Models that receive that same level of polish as core Keras.
 If you are familiar with Keras, congratulations! You already understand most of
-KerasHub.
+KerasNLP.
 
 All models support JAX, TensorFlow, and PyTorch from a single model
 definition and can be fine-tuned on GPUs and TPUs out of the box. Models can
 be trained on individual accelerators with built-in PEFT techniques, or
 fine-tuned at scale with model and data parallel training. See our
-[Getting Started guide](https://keras.io/guides/keras_hub/getting_started)
+[Getting Started guide](https://keras.io/guides/keras_nlp/getting_started)
 to start learning our API. Browse our models on
 [Kaggle](https://www.kaggle.com/organizations/keras/models).
 We welcome contributions.
@@ -27,9 +34,9 @@ We welcome contributions.
 
 ### For everyone
 
-- [Home Page](https://keras.io/keras_hub)
-- [Developer Guides](https://keras.io/guides/keras_hub)
-- [API Reference](https://keras.io/api/keras_hub)
+- [Home Page](https://keras.io/keras_nlp)
+- [Developer Guides](https://keras.io/guides/keras_nlp)
+- [API Reference](https://keras.io/api/keras_nlp)
 - [Pre-trained Models](https://www.kaggle.com/organizations/keras/models)
 
 ### For contributors
@@ -48,7 +55,7 @@ Fine-tune BERT on IMDb movie reviews:
 import os
 os.environ["KERAS_BACKEND"] = "jax"  # Or "tensorflow" or "torch"!
 
-import keras_hub
+import keras_nlp
 import tensorflow_datasets as tfds
 
 imdb_train, imdb_test = tfds.load(
@@ -58,7 +65,7 @@ imdb_train, imdb_test = tfds.load(
     batch_size=16,
 )
 # Load a BERT model.
-classifier = keras_hub.models.Classifier.from_preset(
+classifier = keras_nlp.models.Classifier.from_preset(
     "bert_base_en", 
     num_classes=2,
     activation="softmax",
@@ -71,24 +78,24 @@ classifier.predict(["What an amazing movie!", "A total waste of my time."])
 
 Try it out [in a colab](https://colab.research.google.com/gist/mattdangerw/e457e42d5ea827110c8d5cb4eb9d9a07/kerasnlp-quickstart.ipynb).
 For more in depth guides and examples, visit
-[keras.io/keras_hub](https://keras.io/keras_hub/).
+[keras.io/keras_nlp](https://keras.io/keras_nlp/).
 
 ## Installation
 
-To install the latest KerasHub release with Keras 3, simply run:
+To install the latest KerasNLP release with Keras 3, simply run:
 
 ```
-pip install --upgrade keras-hub
+pip install --upgrade keras-nlp
 ```
 
-To install the latest nightly changes for both KerasHub and Keras, you can use
+To install the latest nightly changes for both KerasNLP and Keras, you can use
 our nightly package.
 
 ```
-pip install --upgrade keras-hub-nightly
+pip install --upgrade keras-nlp-nightly
 ```
 
-Note that currently, installing KerasHub will always pull in TensorFlow for use
+Note that currently, installing KerasNLP will always pull in TensorFlow for use
 of the `tf.data` API for preprocessing. Even when pre-processing with `tf.data`,
 training can still happen on any backend.
 
@@ -96,13 +103,13 @@ Read [Getting started with Keras](https://keras.io/getting_started/) for more
 information on installing Keras 3 and compatibility with different frameworks.
 
 > [!IMPORTANT]
-> We recommend using KerasHub with TensorFlow 2.16 or later, as TF 2.16 packages
+> We recommend using KerasNLP with TensorFlow 2.16 or later, as TF 2.16 packages
 > Keras 3 by default.
 
 ## Configuring your backend
 
 If you have Keras 3 installed in your environment (see installation above),
-you can use KerasHub with any of JAX, TensorFlow and PyTorch. To do so, set the
+you can use KerasNLP with any of JAX, TensorFlow and PyTorch. To do so, set the
 `KERAS_BACKEND` environment variable. For example:
 
 ```shell
@@ -115,7 +122,7 @@ Or in Colab, with:
 import os
 os.environ["KERAS_BACKEND"] = "jax"
 
-import keras_hub
+import keras_nlp
 ```
 
 > [!IMPORTANT]
@@ -131,21 +138,21 @@ may break compatibility at any time and APIs should not be consider stable.
 
 ## Disclaimer
 
-KerasHub provides access to pre-trained models via the `keras_hub.models` API.
+KerasNLP provides access to pre-trained models via the `keras_nlp.models` API.
 These pre-trained models are provided on an "as is" basis, without warranties
 or conditions of any kind. The following underlying models are provided by third
 parties, and subject to separate licenses:
 BART, BLOOM, DeBERTa, DistilBERT, GPT-2, Llama, Mistral, OPT, RoBERTa, Whisper,
 and XLM-RoBERTa.
 
-## Citing KerasHub
+## Citing KerasNLP
 
-If KerasHub helps your research, we appreciate your citations.
+If KerasNLP helps your research, we appreciate your citations.
 Here is the BibTeX entry:
 
 ```bibtex
 @misc{kerasnlp2022,
-  title={KerasHub},
+  title={KerasNLP},
   author={Watson, Matthew, and Qian, Chen, and Bischof, Jonathan and Chollet, 
   Fran\c{c}ois and others},
   year={2022},

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/__init__.py

@@ -11,12 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""DO NOT EDIT.
-
-This file was autogenerated. Do not edit it by hand,
-since your modifications would be overwritten.
-"""
-
 import os
 
 # sentencepiece segfaults on some version of tensorflow if tf is imported first.

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/api/__init__.py

@@ -23,6 +23,7 @@ from keras_hub.api import metrics
 from keras_hub.api import models
 from keras_hub.api import samplers
 from keras_hub.api import tokenizers
+from keras_hub.api import utils
 from keras_hub.src.utils.preset_utils import upload_preset
 from keras_hub.src.version_utils import __version__
 from keras_hub.src.version_utils import version

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/api/models/__init__.py

@@ -25,6 +25,9 @@ from keras_hub.src.models.albert.albert_masked_lm_preprocessor import (
 from keras_hub.src.models.albert.albert_text_classifier import (
     AlbertTextClassifier,
 )
+from keras_hub.src.models.albert.albert_text_classifier import (
+    AlbertTextClassifier as AlbertClassifier,
+)
 from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
     AlbertTextClassifierPreprocessor,
 )
@@ -34,7 +37,6 @@ from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
 from keras_hub.src.models.albert.albert_tokenizer import AlbertTokenizer
 from keras_hub.src.models.backbone import Backbone
 from keras_hub.src.models.bart.bart_backbone import BartBackbone
-from keras_hub.src.models.bart.bart_preprocessor import BartPreprocessor
 from keras_hub.src.models.bart.bart_seq_2_seq_lm import BartSeq2SeqLM
 from keras_hub.src.models.bart.bart_seq_2_seq_lm_preprocessor import (
     BartSeq2SeqLMPreprocessor,
@@ -46,6 +48,9 @@ from keras_hub.src.models.bert.bert_masked_lm_preprocessor import (
     BertMaskedLMPreprocessor,
 )
 from keras_hub.src.models.bert.bert_text_classifier import BertTextClassifier
+from keras_hub.src.models.bert.bert_text_classifier import (
+    BertTextClassifier as BertClassifier,
+)
 from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
     BertTextClassifierPreprocessor,
 )
@@ -58,7 +63,6 @@ from keras_hub.src.models.bloom.bloom_causal_lm import BloomCausalLM
 from keras_hub.src.models.bloom.bloom_causal_lm_preprocessor import (
     BloomCausalLMPreprocessor,
 )
-from keras_hub.src.models.bloom.bloom_preprocessor import BloomPreprocessor
 from keras_hub.src.models.bloom.bloom_tokenizer import BloomTokenizer
 from keras_hub.src.models.causal_lm import CausalLM
 from keras_hub.src.models.causal_lm_preprocessor import CausalLMPreprocessor
@@ -80,6 +84,9 @@ from keras_hub.src.models.deberta_v3.deberta_v3_masked_lm_preprocessor import (
 from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
     DebertaV3TextClassifier,
 )
+from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
+    DebertaV3TextClassifier as DebertaV3Classifier,
+)
 from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
     DebertaV3TextClassifierPreprocessor,
 )
@@ -105,6 +112,9 @@ from keras_hub.src.models.distil_bert.distil_bert_masked_lm_preprocessor import
 from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
     DistilBertTextClassifier,
 )
+from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
+    DistilBertTextClassifier as DistilBertClassifier,
+)
 from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor import (
     DistilBertTextClassifierPreprocessor,
 )
@@ -118,9 +128,6 @@ from keras_hub.src.models.efficientnet.efficientnet_backbone import (
     EfficientNetBackbone,
 )
 from keras_hub.src.models.electra.electra_backbone import ElectraBackbone
-from keras_hub.src.models.electra.electra_preprocessor import (
-    ElectraPreprocessor,
-)
 from keras_hub.src.models.electra.electra_tokenizer import ElectraTokenizer
 from keras_hub.src.models.f_net.f_net_backbone import FNetBackbone
 from keras_hub.src.models.f_net.f_net_masked_lm import FNetMaskedLM
@@ -128,6 +135,9 @@ from keras_hub.src.models.f_net.f_net_masked_lm_preprocessor import (
     FNetMaskedLMPreprocessor,
 )
 from keras_hub.src.models.f_net.f_net_text_classifier import FNetTextClassifier
+from keras_hub.src.models.f_net.f_net_text_classifier import (
+    FNetTextClassifier as FNetClassifier,
+)
 from keras_hub.src.models.f_net.f_net_text_classifier_preprocessor import (
     FNetTextClassifierPreprocessor,
 )
@@ -140,7 +150,6 @@ from keras_hub.src.models.falcon.falcon_causal_lm import FalconCausalLM
 from keras_hub.src.models.falcon.falcon_causal_lm_preprocessor import (
     FalconCausalLMPreprocessor,
 )
-from keras_hub.src.models.falcon.falcon_preprocessor import FalconPreprocessor
 from keras_hub.src.models.falcon.falcon_tokenizer import FalconTokenizer
 from keras_hub.src.models.feature_pyramid_backbone import FeaturePyramidBackbone
 from keras_hub.src.models.gemma.gemma_backbone import GemmaBackbone
@@ -148,7 +157,6 @@ from keras_hub.src.models.gemma.gemma_causal_lm import GemmaCausalLM
 from keras_hub.src.models.gemma.gemma_causal_lm_preprocessor import (
     GemmaCausalLMPreprocessor,
 )
-from keras_hub.src.models.gemma.gemma_preprocessor import GemmaPreprocessor
 from keras_hub.src.models.gemma.gemma_tokenizer import GemmaTokenizer
 from keras_hub.src.models.gpt2.gpt2_backbone import GPT2Backbone
 from keras_hub.src.models.gpt2.gpt2_causal_lm import GPT2CausalLM
@@ -162,9 +170,6 @@ from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm import GPTNeoXCausalLM
 from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm_preprocessor import (
     GPTNeoXCausalLMPreprocessor,
 )
-from keras_hub.src.models.gpt_neo_x.gpt_neo_x_preprocessor import (
-    GPTNeoXPreprocessor,
-)
 from keras_hub.src.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer
 from keras_hub.src.models.image_classifier import ImageClassifier
 from keras_hub.src.models.image_classifier_preprocessor import (
@@ -175,14 +180,12 @@ from keras_hub.src.models.llama3.llama3_causal_lm import Llama3CausalLM
 from keras_hub.src.models.llama3.llama3_causal_lm_preprocessor import (
     Llama3CausalLMPreprocessor,
 )
-from keras_hub.src.models.llama3.llama3_preprocessor import Llama3Preprocessor
 from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
 from keras_hub.src.models.llama.llama_backbone import LlamaBackbone
 from keras_hub.src.models.llama.llama_causal_lm import LlamaCausalLM
 from keras_hub.src.models.llama.llama_causal_lm_preprocessor import (
     LlamaCausalLMPreprocessor,
 )
-from keras_hub.src.models.llama.llama_preprocessor import LlamaPreprocessor
 from keras_hub.src.models.llama.llama_tokenizer import LlamaTokenizer
 from keras_hub.src.models.masked_lm import MaskedLM
 from keras_hub.src.models.masked_lm_preprocessor import MaskedLMPreprocessor
@@ -191,9 +194,6 @@ from keras_hub.src.models.mistral.mistral_causal_lm import MistralCausalLM
 from keras_hub.src.models.mistral.mistral_causal_lm_preprocessor import (
     MistralCausalLMPreprocessor,
 )
-from keras_hub.src.models.mistral.mistral_preprocessor import (
-    MistralPreprocessor,
-)
 from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
 from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
     MiTBackbone,
@@ -210,7 +210,6 @@ from keras_hub.src.models.opt.opt_causal_lm import OPTCausalLM
 from keras_hub.src.models.opt.opt_causal_lm_preprocessor import (
     OPTCausalLMPreprocessor,
 )
-from keras_hub.src.models.opt.opt_preprocessor import OPTPreprocessor
 from keras_hub.src.models.opt.opt_tokenizer import OPTTokenizer
 from keras_hub.src.models.pali_gemma.pali_gemma_backbone import (
     PaliGemmaBackbone,
@@ -229,7 +228,6 @@ from keras_hub.src.models.phi3.phi3_causal_lm import Phi3CausalLM
 from keras_hub.src.models.phi3.phi3_causal_lm_preprocessor import (
     Phi3CausalLMPreprocessor,
 )
-from keras_hub.src.models.phi3.phi3_preprocessor import Phi3Preprocessor
 from keras_hub.src.models.phi3.phi3_tokenizer import Phi3Tokenizer
 from keras_hub.src.models.preprocessor import Preprocessor
 from keras_hub.src.models.resnet.resnet_backbone import ResNetBackbone
@@ -247,6 +245,9 @@ from keras_hub.src.models.roberta.roberta_masked_lm_preprocessor import (
 from keras_hub.src.models.roberta.roberta_text_classifier import (
     RobertaTextClassifier,
 )
+from keras_hub.src.models.roberta.roberta_text_classifier import (
+    RobertaTextClassifier as RobertaClassifier,
+)
 from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
     RobertaTextClassifierPreprocessor,
 )
@@ -260,6 +261,7 @@ from keras_hub.src.models.t5.t5_backbone import T5Backbone
 from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer
 from keras_hub.src.models.task import Task
 from keras_hub.src.models.text_classifier import TextClassifier
+from keras_hub.src.models.text_classifier import TextClassifier as Classifier
 from keras_hub.src.models.text_classifier_preprocessor import (
     TextClassifierPreprocessor,
 )
@@ -280,6 +282,9 @@ from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm_preprocessor import
 from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
     XLMRobertaTextClassifier,
 )
+from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
+    XLMRobertaTextClassifier as XLMRobertaClassifier,
+)
 from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
     XLMRobertaTextClassifierPreprocessor,
 )

keras_hub_nightly-0.16.0.dev2024092017/keras_hub/api/utils/__init__.py

@@ -0,0 +1,22 @@
+# Copyright 2024 The KerasHub Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""DO NOT EDIT.
+
+This file was autogenerated. Do not edit it by hand,
+since your modifications would be overwritten.
+"""
+
+from keras_hub.src.utils.imagenet.imagenet_utils import (
+    decode_imagenet_predictions,
+)

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/api_export.py

@@ -14,7 +14,7 @@
 
 import types
 
-import keras
+from keras.saving import register_keras_serializable
 
 try:
     import namex
@@ -22,14 +22,20 @@ except ImportError:
     namex = None
 
 
-def maybe_register_serializable(symbol):
+def maybe_register_serializable(path, symbol):
+    if isinstance(path, (list, tuple)):
+        # If we have multiple export names, actually make sure to register these
+        # first. This makes sure we have a backward compat mapping of old
+        # serialized names to new class.
+        for name in path:
+            name = name.split(".")[-1]
+            register_keras_serializable(package="keras_nlp", name=name)(symbol)
+            register_keras_serializable(package="keras_hub", name=name)(symbol)
     if isinstance(symbol, types.FunctionType) or hasattr(symbol, "get_config"):
-        # We register twice, first with the old name, second with the new name,
-        # so loading still works under the old name.
-        # TODO replace keras_nlp with keras-hub after rename.
-        compat_name = "keras_nlp"
-        keras.saving.register_keras_serializable(package=compat_name)(symbol)
-        keras.saving.register_keras_serializable(package="keras_hub")(symbol)
+        # We register twice, first with keras_nlp, second with keras_hub,
+        # so loading still works for classes saved as "keras_nlp".
+        register_keras_serializable(package="keras_nlp")(symbol)
+        register_keras_serializable(package="keras_hub")(symbol)
 
 
 if namex:
@@ -39,7 +45,7 @@ if namex:
             super().__init__(package="keras_hub", path=path)
 
         def __call__(self, symbol):
-            maybe_register_serializable(symbol)
+            maybe_register_serializable(self.path, symbol)
             return super().__call__(symbol)
 
 else:

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/albert/albert_text_classifier.py

@@ -25,7 +25,12 @@ from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
 from keras_hub.src.models.text_classifier import TextClassifier
 
 
-@keras_hub_export("keras_hub.models.AlbertTextClassifier")
+@keras_hub_export(
+    [
+        "keras_hub.models.AlbertTextClassifier",
+        "keras_hub.models.AlbertClassifier",
+    ]
+)
 class AlbertTextClassifier(TextClassifier):
     """An end-to-end ALBERT model for classification tasks
 

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bert/bert_text_classifier.py

@@ -23,7 +23,12 @@ from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
 from keras_hub.src.models.text_classifier import TextClassifier
 
 
-@keras_hub_export("keras_hub.models.BertTextClassifier")
+@keras_hub_export(
+    [
+        "keras_hub.models.BertTextClassifier",
+        "keras_hub.models.BertClassifier",
+    ]
+)
 class BertTextClassifier(TextClassifier):
     """An end-to-end BERT model for classification tasks.
 

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/deberta_v3_text_classifier.py

@@ -28,7 +28,12 @@ from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor imp
 from keras_hub.src.models.text_classifier import TextClassifier
 
 
-@keras_hub_export("keras_hub.models.DebertaV3TextClassifier")
+@keras_hub_export(
+    [
+        "keras_hub.models.DebertaV3TextClassifier",
+        "keras_hub.models.DebertaV3Classifier",
+    ]
+)
 class DebertaV3TextClassifier(TextClassifier):
     """An end-to-end DeBERTa model for classification tasks.
 

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/densenet/densenet_backbone.py

@@ -114,7 +114,7 @@ class DenseNetBackbone(FeaturePyramidBackbone):
             growth_rate,
             name=f"conv{len(stackwise_num_repeats) + 1}",
         )
-        pyramid_outputs[f"P{len(stackwise_num_repeats) +1}"] = x
+        pyramid_outputs[f"P{len(stackwise_num_repeats) + 1}"] = x
         x = keras.layers.BatchNormalization(
             axis=channel_axis, epsilon=BN_EPSILON, name="bn"
         )(x)

{keras_hub_nightly-0.15.0.dev20240911134614 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/distil_bert/distil_bert_text_classifier.py

@@ -28,7 +28,12 @@ from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor i
 from keras_hub.src.models.text_classifier import TextClassifier
 
 
-@keras_hub_export("keras_hub.models.DistilBertTextClassifier")
+@keras_hub_export(
+    [
+        "keras_hub.models.DistilBertTextClassifier",
+        "keras_hub.models.DistilBertClassifier",
+    ]
+)
 class DistilBertTextClassifier(TextClassifier):
     """An end-to-end DistilBERT model for classification tasks.