From fb528b60c68a2f592bef492fd4b2d7dc928865ab Mon Sep 17 00:00:00 2001 From: Chun-Hsiang Wang Date: Wed, 3 Aug 2022 15:23:35 -0700 Subject: [PATCH] samples: Minor fix for CPR existing image use cases. (#778) --- .../SDK_Triton_PyTorch_Local_Prediction.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/community/prediction/custom_prediction_routines/SDK_Triton_PyTorch_Local_Prediction.ipynb b/notebooks/community/prediction/custom_prediction_routines/SDK_Triton_PyTorch_Local_Prediction.ipynb index ec6aba9db..4fb8e5b3c 100644 --- a/notebooks/community/prediction/custom_prediction_routines/SDK_Triton_PyTorch_Local_Prediction.ipynb +++ b/notebooks/community/prediction/custom_prediction_routines/SDK_Triton_PyTorch_Local_Prediction.ipynb @@ -967,7 +967,7 @@ "source": [ "from google.cloud.aiplatform.prediction import LocalModel\n", "\n", - "local_model = LocalModel.create(\n", + "local_model = LocalModel(\n", " serving_container_image_uri=f\"nvcr.io/nvidia/tritonserver:{TRITON_VERSION}-py3\",\n", " serving_container_predict_route=f\"/v2/models/{MODEL_NAME}/infer\",\n", " serving_container_health_route=f\"/v2/models/{MODEL_NAME}\",\n",