fix requirement and setup

4480c51b · kinghuin · wuzewu · c2c1427c · 4480c51b · 4480c51b
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

requirements.txt requirements.txt +1 -0

requirements_py2.txt requirements_py2.txt +1 -0

setup.py setup.py +2 -1

未找到文件。
--- a/requirements.txt
+++ b/requirements.txt
@@ -14,3 +14,4 @@ flake8
 tb-paddle
 tensorboard >= 1.15
 cma == 2.7.0
+sentencepiece
--- 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
--- 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: