diff --git a/model-mesh-mlserver-adapter/server/server.go b/model-mesh-mlserver-adapter/server/server.go
index 57047f94..2d28d13d 100644
--- a/model-mesh-mlserver-adapter/server/server.go
+++ b/model-mesh-mlserver-adapter/server/server.go
@@ -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