diff --git a/requirements.txt b/requirements.txt index 5866e150bb2caf45ddd6804cba4598a0c899c5dd..9264b7829c6ad275a7288c9724e5f5014a13d805 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,3 +14,4 @@ flake8 tb-paddle tensorboard >= 1.15 cma == 2.7.0 +sentencepiece diff --git a/requirements_py2.txt b/requirements_py2.txt index e7e54a3fe3b6c9bad7da52df4878d74a14730ecc..c3981c3a54a4f36922854587c137af448b19350c 100644 --- a/requirements_py2.txt +++ b/requirements_py2.txt @@ -13,3 +13,4 @@ flask >= 1.1.0 tb-paddle tensorboard >= 1.15 cma == 2.7.0 +sentencepiece diff --git a/setup.py b/setup.py index 8a7d52461f1976a742b564f37f32fcc03dfab672..4a5cc342a2e83c1dd004d71af894189cff913bbf 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,8 @@ max_version, mid_version, min_version = python_version() REQUIRED_PACKAGES = [ 'six >= 1.10.0', 'protobuf >= 3.6.0', 'pyyaml', 'Pillow', 'requests', - 'tb-paddle', 'tensorboard >= 1.15', 'cma == 2.7.0', 'flask >= 1.1.0' + 'tb-paddle', 'tensorboard >= 1.15', 'cma == 2.7.0', 'flask >= 1.1.0', + 'sentencepiece' ] if max_version < 3: