diff --git a/setup.py b/setup.py index e6e79a1e..6db4dad9 100644 --- a/setup.py +++ b/setup.py @@ -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', ] -) \ No newline at end of file +)