hydra/omegaconf required

yzhangcs · yzhangcs · commit 46cf49b3605e · 2022-05-27T08:53:39.000+08:00
diff --git a/setup.py b/setup.py
@@ -26,8 +26,10 @@
         'numpy<1.21.5; python_version<"3.8"',
         'torch>=1.7.1',
         'transformers>=4.0.0',
+        'hydra-core>=1.2',
         'nltk',
         'stanza',
+        'omegaconf',
         'dill'],
     extras_require={
         'elmo': ['allennlp']
