SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[ci] Pin torch version while we update

Browse Source
This commit is contained in:
Julien Chaumond
2020-04-21 15:46:18 -04:00
parent 53f5ef6df5
commit eb5601b0a5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -67,7 +67,7 @@ extras["mecab"] = ["mecab-python3"]
extras["sklearn"] = ["scikit-learn"] extras["sklearn"] = ["scikit-learn"]
extras["tf"] = ["tensorflow"] extras["tf"] = ["tensorflow"]
extras["tf-cpu"] = ["tensorflow-cpu"] extras["tf-cpu"] = ["tensorflow-cpu"]
extras["torch"] = ["torch"] extras["torch"] = ["torch==1.4.0"]
extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"] extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"]
extras["all"] = extras["serving"] + ["tensorflow", "torch"] extras["all"] = extras["serving"] + ["tensorflow", "torch"]