diff --git a/requirements.txt b/requirements.txt index 552b1ec299..7b59e73759 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ # Tensorflow. tensorflow-cpu~=2.18.0;sys_platform != 'darwin' tensorflow~=2.18.0;sys_platform == 'darwin' -tensorflow-text~=2.18 +tensorflow-text~=2.18;platform_system != 'Windows' # Torch. --extra-index-url https://download.pytorch.org/whl/cpu diff --git a/setup.py b/setup.py index b9f1588039..138e295af8 100644 --- a/setup.py +++ b/setup.py @@ -45,13 +45,13 @@ def get_version(rel_path): "regex", "rich", "kagglehub", + "tensorflow-text;platform_system != 'Windows'", ], extras_require={ "extras": [ "rouge-score", "sentencepiece", ], - "nlp": ["tensorflow-text"], }, # Supported Python versions python_requires=">=3.9",