xiaohei/model_garden
1
0
Fork 0
mirror of https://github.com/GoogleCloudPlatform/vertex-ai-samples.git synced 2026-09-26 14:42:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

samples: Minor fix for CPR existing image use cases. (#778)

Browse Source
This commit is contained in:
Chun-Hsiang Wang
2022-08-03 15:23:35 -07:00
committed by GitHub
parent 1c1a2b2097
commit fb528b60c6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/community/prediction/custom_prediction_routines/SDK_Triton_PyTorch_Local_Prediction.ipynb
+1 -1
View File
@@ -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",