set minimum safe tensorflow

minimaxir · Oct 18, 2021 · d1e97f5 · d1e97f5
1 parent 75426fc
commit d1e97f5
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -24,7 +24,7 @@ gpt-2-simple can be installed [via PyPI](https://pypi.org/project/gpt_2_simple/)
 pip3 install gpt-2-simple
 ```
 
-You will also need to install the corresponding TensorFlow 2.X version for your system (e.g. `tensorflow` or `tensorflow-gpu`).
+You will also need to install the corresponding TensorFlow 2.X version (min 2.5.1) for your system (e.g. `tensorflow` or `tensorflow-gpu`).
 
 ## Usage
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-tensorflow==1.15.2
+tensorflow>=2.5.1
 regex
 requests
 tqdm

diff --git a/setup.py b/setup.py
@@ -47,7 +47,7 @@
 setup(
     name="gpt_2_simple",
     packages=["gpt_2_simple"],  # this must be the same as the name above
-    version="0.8.0",
+    version="0.8.1",
     description="Python package to easily retrain OpenAI's GPT-2 "
     "text-generating model on new texts.",
     long_description=long_description,
@@ -63,5 +63,12 @@
     },
     python_requires=">=3.6",
     include_package_data=True,
-    install_requires=["regex", "requests", "tqdm", "numpy", "toposort"],
+    install_requires=[
+        "tensorflow>=2.5.1",
+        "regex",
+        "requests",
+        "tqdm",
+        "numpy",
+        "toposort",
+    ],
 )