diff --git a/README.md b/README.md index 4a5aa35899..37870a237d 100644 --- a/README.md +++ b/README.md @@ -102,6 +102,13 @@ To install the latest KerasHub release with Keras 3, simply run: pip install --upgrade keras-hub ``` +Our text tokenizers are based on TensorFlow Text. Hence, if you are using any +model which has language as a modality, you will have to run: + +``` +pip install --upgrade keras-hub[nlp] +``` + To install the latest nightly changes for both KerasHub and Keras, you can use our nightly package. diff --git a/setup.py b/setup.py index 6495f56bf5..b9f1588039 100644 --- a/setup.py +++ b/setup.py @@ -45,13 +45,13 @@ def get_version(rel_path): "regex", "rich", "kagglehub", - "tensorflow-text", ], extras_require={ "extras": [ "rouge-score", "sentencepiece", ], + "nlp": ["tensorflow-text"], }, # Supported Python versions python_requires=">=3.9",