Necessary pkgs for ELMo

yzhangcs · yzhangcs · commit b0b7204a9961 · 2021-07-17T18:32:09.000+08:00
diff --git a/setup.py b/setup.py
@@ -25,10 +25,12 @@
     install_requires=[
         'torch>=1.7.1',
         'transformers>=4.0.0',
-        'allennlp',
         'nltk',
         'stanza',
         'dill'],
+    extras_require={
+        'elmo': ['allennlp']
+    },
     entry_points={
         'console_scripts': [
             'biaffine-dep=supar.cmds.biaffine_dep:main',
