fix: requirements in nemo configs + add torchvision to pytorch.txt (#…

…1186) * fix: requirements in nemo configs + added torchvision to pytorch.txt * refactor: latest pytorch and torchvision
deeppavlov · Apr 27, 2020 · 5f5bbbd · 5f5bbbd
1 parent fdab459
commit 5f5bbbd
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 2 deletions.
diff --git a/deeppavlov/__init__.py b/deeppavlov/__init__.py
@@ -39,7 +39,7 @@ def evaluate_model(config: [str, Path, dict], download: bool = False, recursive:
 except ImportError:
     'Assuming that requirements are not yet installed'
 
-__version__ = '0.9.0'
+__version__ = '0.9.1'
 __author__ = 'Neural Networks and Deep Learning lab, MIPT'
 __description__ = 'An open source library for building end-to-end dialog systems and training chatbots.'
 __keywords__ = ['NLP', 'NER', 'SQUAD', 'Intents', 'Chatbot']

diff --git a/deeppavlov/configs/nemo/asr.json b/deeppavlov/configs/nemo/asr.json
@@ -17,6 +17,7 @@
       "NEMO_PATH": "~/.deeppavlov/models/nemo"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt"
     ],
     "download": [

diff --git a/deeppavlov/configs/nemo/asr_tts.json b/deeppavlov/configs/nemo/asr_tts.json
@@ -35,6 +35,7 @@
       "TTS_PATH": "{NEMO_PATH}/tacotron2_waveglow"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-tts.txt"
     ],

diff --git a/deeppavlov/configs/nemo/tts.json b/deeppavlov/configs/nemo/tts.json
@@ -18,6 +18,7 @@
       "TTS_PATH": "{NEMO_PATH}/tacotron2_waveglow"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-tts.txt"
     ],

diff --git a/deeppavlov/requirements/pytorch.txt b/deeppavlov/requirements/pytorch.txt
@@ -1 +1,2 @@
-torch==1.4.0
+torch==1.5.0
+torchvision==0.6.0
diff --git a/tests/test_configs/nemo/tts2asr_test.json b/tests/test_configs/nemo/tts2asr_test.json
@@ -36,6 +36,7 @@
       "TTS_PATH": "{NEMO_PATH}/tacotron2_waveglow"
     },
     "requirements": [
+      "{DEEPPAVLOV_PATH}/requirements/pytorch.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-asr.txt",
       "{DEEPPAVLOV_PATH}/requirements/nemo-tts.txt"
     ],