dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Provide serving dependencies for tensorflow and pytorch (serving-tf, serving-torch)

This commit is contained in:
Morgan Funtowicz 2019-12-17 11:23:39 +01:00
parent a096e2a88b
commit d7c62661a3
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

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