Update requirements

jamestiotio · Nov 16, 2022 · 601b3aa · 601b3aa
1 parent cfcbc66
commit 601b3aa
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 17 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,8 +1,10 @@
-fire==0.4.0
-ipython==8.4.0
-numpy==1.22.0
-pandas==1.3.5
-pyzmq==23.2.1
-regex==2022.3.15
-setuptools==58.0.4
-tqdm==4.63.0
+fire>=0.4.0
+ipython>=8.4.0
+numpy>=1.22.0
+pandas>=1.3.5
+pyzmq>=23.2.1
+regex>=2022.3.15
+setuptools>=58.0.4
+transformers>=4.22.0
+pytorch>=1.10.0
+tqdm>=4.63.0
diff --git a/setup.py b/setup.py
@@ -4,18 +4,20 @@
     name="codegeex",
     py_modules=["codegeex"],
     version="1.0",
-    description="CodeGeeX: A Multilingual Code Generative Model.",
+    description="CodeGeeX: A Open Multilingual Code Generation Model.",
     author="Qinkai Zheng",
     packages=find_packages(),
     install_requires=[
-        "fire==0.4.0",
-        "ipython==8.4.0",
-        "numpy==1.22.0",
-        "pandas==1.3.5",
-        "pyzmq==23.2.1",
-        "regex==2022.3.15",
-        "setuptools==58.0.4",
-        "tqdm==4.63.0",
+        "fire>=0.4.0",
+        "ipython>=8.4.0",
+        "numpy>=1.22.0",
+        "pandas>=1.3.5",
+        "pyzmq>=23.2.1",
+        "regex>=2022.3.15",
+        "setuptools>=58.0.4",
+        "transformers>=4.22.0",
+        "pytorch>=1.10.0",
+        "tqdm>=4.63.0",
     ],
     entry_points={}
 )