Requires torch>=1.8 and pathos

yzhangcs · yzhangcs · commit 95680c6790c6 · 2022-06-14T21:04:17.000+08:00
diff --git a/README.md b/README.md
@@ -41,7 +41,7 @@ $ pip install -U git+https://github.com/yzhangcs/parser
 
 As a prerequisite, the following requirements should be satisfied:
 * `python`: >= 3.7
-* [`pytorch`](https://github.com/pytorch/pytorch): >= 1.7
+* [`pytorch`](https://github.com/pytorch/pytorch): >= 1.8
 * [`transformers`](https://github.com/huggingface/transformers): >= 4.0
 
 ## Usage
diff --git a/setup.py b/setup.py
@@ -25,13 +25,14 @@
     ],
     install_requires=[
         'numpy<1.21.5; python_version<"3.8"',
-        'torch>=1.7.1',
+        'torch>=1.8',
         'transformers>=4.0.0',
         'hydra-core>=1.2',
         'nltk',
         'stanza',
         'omegaconf',
-        'dill'],
+        'dill',
+        'pathos'],
     extras_require={
         'elmo': ['allennlp']
     },
