Use binary format by default for now

triton-inference-server · Oct 2, 2024 · 240b653 · 240b653
1 parent 90ae57d
commit 240b653
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/qa/L0_e2e/generate_example_model.py b/qa/L0_e2e/generate_example_model.py
@@ -467,7 +467,9 @@ def build_model(
 
     if output_format is None:
         if model_type == "xgboost":
-            output_format = "xgboost_ubj"
+            # TODO(hcho3): Update to "xgboost_ubj" when XGBoost removes support
+            # for legacy binary format
+            output_format = "xgboost"
         elif model_type == "lightgbm":
             output_format = "lightgbm"
         elif model_type in {"sklearn", "cuml"}: