Keep TF Text as optional dependency (#2103)

keras-team · Feb 19, 2025 · 899805e · 899805e
1 parent 97fc29c
commit 899805e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git 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
@@ -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",