Skip to content

Commit

Permalink
feat: Add mlflow in model implementation mapping
Browse files Browse the repository at this point in the history
The `mlflow` model format is supported by mlserver in kserver serverless
mode, and modelmesh too. Without the mapping, mlserver fails to load the
model.
  • Loading branch information
Erouan50 committed Aug 28, 2024
1 parent 5c07033 commit d11a094
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions model-mesh-mlserver-adapter/server/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -369,6 +369,7 @@ func generateModelConfigJSON(modelID string, modelType string, uri string, schem
"sklearn": "mlserver_sklearn.SKLearnModel",
"xgboost": "mlserver_xgboost.XGBoostModel",
"mllib": "mlserver-mllib.MLlibModel",
"mlflow": "mlserver_mlflow.MLflowRuntime",
}

// set the implementation
Expand Down

0 comments on commit d11a094

Please sign in to comment.