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",