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

update serving command

Browse Source
This commit is contained in:
thomwolf
2019-12-20 13:47:35 +01:00
parent 15dda5ea32
commit 73fcebf7ec
3 changed files with 27 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
setup.py
View File

@@ -38,9 +38,9 @@ from setuptools import find_packages, setup
extras = {
'serving': ['uvicorn', 'fastapi'],
'serving-tf': ['uvicorn', 'fastapi', 'tensorflow'],
'serving-torch': ['uvicorn', 'fastapi', 'torch']
'serving': ['pydantic', 'uvicorn', 'fastapi'],
'serving-tf': ['pydantic', 'uvicorn', 'fastapi', 'tensorflow'],
'serving-torch': ['pydantic', 'uvicorn', 'fastapi', 'torch']
}
extras['all'] = [package for package in extras.values()]