add required package

This commit is contained in:
Jim Deng 2021-01-15 02:13:53 +00:00
parent e680cbc2da
commit ea0c9dedb5
1 changed files with 2 additions and 1 deletions

View File

@ -58,9 +58,10 @@ setuptools.setup(
'sentence-transformers==0.4.0',
'sentencepiece==0.1.*',
'loss_dropper @ git+https://github.com/ddkang/loss_dropper#egg=loss_dropper',
'kfserving>=0.5.0',
# for calibration:
'scikit-learn~=0.23',
'dill~=0.3',
'xgboost~=1.3',
]
)
)