From eb5601b0a5a88824a2598956f96e06e7f2422bce Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Tue, 21 Apr 2020 15:46:18 -0400 Subject: [PATCH] [ci] Pin torch version while we update --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 1e6fa26456..782b122f88 100644 --- a/setup.py +++ b/setup.py @@ -67,7 +67,7 @@ extras["mecab"] = ["mecab-python3"] extras["sklearn"] = ["scikit-learn"] extras["tf"] = ["tensorflow"] extras["tf-cpu"] = ["tensorflow-cpu"] -extras["torch"] = ["torch"] +extras["torch"] = ["torch==1.4.0"] extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"] extras["all"] = extras["serving"] + ["tensorflow", "torch"]