Merge pull request #684 from rajeevsrao/master

Use torch 1.4.0 for exporting Tacotron2 ONNX models
NVIDIA · Jul 13, 2020 · e32ba1c · e32ba1c
2 parents ead9e8b + 5e7c9ba
commit e32ba1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/Tacotron2/scripts/install_prerequisites.sh b/demo/Tacotron2/scripts/install_prerequisites.sh
@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-pip3 install numba==0.48 torch==1.5.1
+pip3 install numba==0.48 torch==1.4.0
 pip3 install -r requirements.txt
 sudo apt-get install -y libsndfile1