diff --git a/config/overlays/test/configmap/inferenceservice.yaml b/config/overlays/test/configmap/inferenceservice.yaml index 4b5952c2ca6..c2c02a6b261 100644 --- a/config/overlays/test/configmap/inferenceservice.yaml +++ b/config/overlays/test/configmap/inferenceservice.yaml @@ -7,18 +7,12 @@ data: predictors: |- { "tensorflow": { - "image": "tensorflow/serving", - "defaultImageVersion": "1.14.0", - "defaultGpuImageVersion": "1.14.0-gpu", - "defaultTimeout": "60", "supportedFrameworks": [ "tensorflow" ], "multiModelServer": false }, "onnx": { - "image": "mcr.microsoft.com/onnxruntime/server", - "defaultImageVersion": "v1.0.0", "supportedFrameworks": [ "onnx" ], @@ -26,16 +20,12 @@ data: }, "sklearn": { "v1": { - "image": "809251082950.dkr.ecr.us-west-2.amazonaws.com/kserve/sklearnserver", - "defaultImageVersion": "latest", "supportedFrameworks": [ "sklearn" ], "multiModelServer": true }, "v2": { - "image": "docker.io/seldonio/mlserver", - "defaultImageVersion": "0.5.3", "supportedFrameworks": [ "sklearn" ], @@ -44,16 +34,12 @@ data: }, "xgboost": { "v1": { - "image": "809251082950.dkr.ecr.us-west-2.amazonaws.com/kserve/xgbserver", - "defaultImageVersion": "latest", "supportedFrameworks": [ "xgboost" ], "multiModelServer": true }, "v2": { - "image": "docker.io/seldonio/mlserver", - "defaultImageVersion": "0.5.3", "supportedFrameworks": [ "xgboost" ], @@ -61,25 +47,18 @@ data: } }, "pytorch": { - "image": "kserve/torchserve-kfs", - "defaultImageVersion": "0.5.2", - "defaultGpuImageVersion": "0.5.2-gpu", "supportedFrameworks": [ "pytorch" ], "multiModelServer": false }, "paddle": { - "image": "ruminateer/paddleserver", - "defaultImageVersion": "v0.1.0", "supportedFrameworks": [ "paddle" ], "multiModelServer": false }, "triton": { - "image": "nvcr.io/nvidia/tritonserver", - "defaultImageVersion": "20.08-py3", "supportedFrameworks": [ "tensorrt", "tensorflow", @@ -90,16 +69,12 @@ data: "multiModelServer": true }, "pmml": { - "image": "809251082950.dkr.ecr.us-west-2.amazonaws.com/kserve/pmmlserver", - "defaultImageVersion": "latest", "supportedFrameworks": [ "pmml" ], "multiModelServer": false }, "lightgbm": { - "image": "809251082950.dkr.ecr.us-west-2.amazonaws.com/kserve/lgbserver", - "defaultImageVersion": "latest", "supportedFrameworks": [ "lightgbm" ], diff --git a/pkg/apis/serving/v1beta1/inference_service_defaults.go b/pkg/apis/serving/v1beta1/inference_service_defaults.go index 23a7ab970b7..5c843b316ab 100644 --- a/pkg/apis/serving/v1beta1/inference_service_defaults.go +++ b/pkg/apis/serving/v1beta1/inference_service_defaults.go @@ -371,6 +371,8 @@ func (isvc *InferenceService) setMlServerDefaults() { modelClass := constants.MLServerModelClassSKLearn if isvc.Spec.Predictor.Model.ModelFormat.Name == constants.SupportedModelXGBoost { modelClass = constants.MLServerModelClassXGBoost + } else if isvc.Spec.Predictor.Model.ModelFormat.Name == constants.SupportedModelLightGBM { + modelClass = constants.MLServerModelClassLightGBM } if isvc.ObjectMeta.Labels == nil { isvc.ObjectMeta.Labels = map[string]string{constants.ModelClassLabel: modelClass} diff --git a/pkg/constants/constants.go b/pkg/constants/constants.go index 936d38584c4..c720e005364 100644 --- a/pkg/constants/constants.go +++ b/pkg/constants/constants.go @@ -292,6 +292,7 @@ const ( const ( MLServerModelClassSKLearn = "mlserver_sklearn.SKLearnModel" MLServerModelClassXGBoost = "mlserver_xgboost.XGBoostModel" + MLServerModelClassLightGBM = "mlserver_lightgbm.LightGBMModel" ) // torchserve service envelope label allowed values diff --git a/python/lgbserver/setup.py b/python/lgbserver/setup.py index 0544c56d3c0..304562f08e8 100644 --- a/python/lgbserver/setup.py +++ b/python/lgbserver/setup.py @@ -34,10 +34,9 @@ packages=find_packages("lgbserver"), install_requires=[ "kserve>=0.7.0", - "lightgbm == 2.3.1", + "lightgbm == 3.3.2", "pandas == 0.25.3", "argparse >= 1.4.0", - "numpy == 1.19.5", ], tests_require=tests_require, extras_require={'test': tests_require}