diff --git a/notebooks/community/model_garden/model_garden_e5.ipynb b/notebooks/community/model_garden/model_garden_e5.ipynb index 2a92ba149..59b500568 100644 --- a/notebooks/community/model_garden/model_garden_e5.ipynb +++ b/notebooks/community/model_garden/model_garden_e5.ipynb @@ -211,6 +211,7 @@ " serving_container_args=tei_args,\n", " serving_container_ports=[7080],\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/intfloat/models/e5\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_hf_paligemma2_deployment.ipynb b/notebooks/community/model_garden/model_garden_hf_paligemma2_deployment.ipynb index 4d7307049..c16d09fe6 100644 --- a/notebooks/community/model_garden/model_garden_hf_paligemma2_deployment.ipynb +++ b/notebooks/community/model_garden/model_garden_hf_paligemma2_deployment.ipynb @@ -179,6 +179,7 @@ " serving_container_predict_route=serving_route,\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/paligemma\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_jax_dito.ipynb b/notebooks/community/model_garden/model_garden_jax_dito.ipynb index b83acf15f..63ebb2a56 100644 --- a/notebooks/community/model_garden/model_garden_jax_dito.ipynb +++ b/notebooks/community/model_garden/model_garden_jax_dito.ipynb @@ -1328,6 +1328,7 @@ " serving_container_args=[],\n", " location=REGION,\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/dito\"\n", ")\n", "\n", "jax_dito_endpoint = jax_dito_model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_jax_fvlm.ipynb b/notebooks/community/model_garden/model_garden_jax_fvlm.ipynb index 4357a00bb..af8f43a78 100644 --- a/notebooks/community/model_garden/model_garden_jax_fvlm.ipynb +++ b/notebooks/community/model_garden/model_garden_jax_fvlm.ipynb @@ -1330,6 +1330,7 @@ " serving_container_args=[],\n", " location=REGION,\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/f-vlm-jax\"\n", ")\n", "\n", "jax_fvlm_endpoint = jax_fvlm_model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_jax_owl_vit_v2.ipynb b/notebooks/community/model_garden/model_garden_jax_owl_vit_v2.ipynb index 01fb3d6a7..6d314a04f 100644 --- a/notebooks/community/model_garden/model_garden_jax_owl_vit_v2.ipynb +++ b/notebooks/community/model_garden/model_garden_jax_owl_vit_v2.ipynb @@ -327,6 +327,7 @@ " serving_container_args=[],\n", " location=REGION,\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/jax-owl-vit-v2\",\n", ")\n", "\n", "endpoints[\"jax_owl2_endpoint\"] = models[\"jax_owl2_model\"].deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_jax_paligemma_deployment.ipynb b/notebooks/community/model_garden/model_garden_jax_paligemma_deployment.ipynb index 2d27e07f8..70ab1848f 100644 --- a/notebooks/community/model_garden/model_garden_jax_paligemma_deployment.ipynb +++ b/notebooks/community/model_garden/model_garden_jax_paligemma_deployment.ipynb @@ -377,6 +377,7 @@ " \"RESOLUTION\": resolution,\n", " \"MODEL_ID\": \"google/\" + model_name,\n", " },\n", + " model_garden_source_model_name=\"publishers/google/models/paligemma\",\n", " )\n", " print(\n", " f\"Deploying {model_name_with_time} on {machine_type} with {accelerator_count} {accelerator_type} GPU(s).\"\n", diff --git a/notebooks/community/model_garden/model_garden_jax_paligemma_finetuning.ipynb b/notebooks/community/model_garden/model_garden_jax_paligemma_finetuning.ipynb index 29a654010..854fcbe8a 100644 --- a/notebooks/community/model_garden/model_garden_jax_paligemma_finetuning.ipynb +++ b/notebooks/community/model_garden/model_garden_jax_paligemma_finetuning.ipynb @@ -556,6 +556,7 @@ " \"RESOLUTION\": resolution,\n", " \"MODEL_ID\": model_name,\n", " },\n", + " model_garden_source_model_name=\"publishers/google/models/paligemma\",\n", " )\n", " print(\n", " f\"Deploying {model_name_with_time} on {machine_type} with {accelerator_count} {accelerator_type} GPU(s).\"\n", diff --git a/notebooks/community/model_garden/model_garden_jax_stable_diffusion_xl.ipynb b/notebooks/community/model_garden/model_garden_jax_stable_diffusion_xl.ipynb index 6981ad052..cf6537688 100644 --- a/notebooks/community/model_garden/model_garden_jax_stable_diffusion_xl.ipynb +++ b/notebooks/community/model_garden/model_garden_jax_stable_diffusion_xl.ipynb @@ -275,6 +275,7 @@ " serving_container_ports=[8080],\n", " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-xl-base\"\n", " )\n", " machine_type = \"ct5lp-hightpu-1t\"\n", "\n", diff --git a/notebooks/community/model_garden/model_garden_jax_vision_transformer.ipynb b/notebooks/community/model_garden/model_garden_jax_vision_transformer.ipynb index 710514581..599d1080e 100644 --- a/notebooks/community/model_garden/model_garden_jax_vision_transformer.ipynb +++ b/notebooks/community/model_garden/model_garden_jax_vision_transformer.ipynb @@ -785,6 +785,7 @@ " serving_container_args=[],\n", " location=REGION,\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/vit-jax\"\n", ")\n", "\n", "jax_vit_endpoint = jax_vit_model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_keras_stable_diffusion.ipynb b/notebooks/community/model_garden/model_garden_keras_stable_diffusion.ipynb index 5622b2012..eb5d1226c 100644 --- a/notebooks/community/model_garden/model_garden_keras_stable_diffusion.ipynb +++ b/notebooks/community/model_garden/model_garden_keras_stable_diffusion.ipynb @@ -360,6 +360,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/stable-diffusion-v1-4\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_keras_yolov8.ipynb b/notebooks/community/model_garden/model_garden_keras_yolov8.ipynb index 15c794597..e779cc432 100644 --- a/notebooks/community/model_garden/model_garden_keras_yolov8.ipynb +++ b/notebooks/community/model_garden/model_garden_keras_yolov8.ipynb @@ -508,6 +508,7 @@ " serving_container_image_uri=SERVING_CONTAINER_URI,\n", " serving_container_args=SERVING_CONTAINER_ARGS,\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/keras-yolov8\",\n", ")\n", "\n", "print(\"The model name is: \", upload_job_name)" diff --git a/notebooks/community/model_garden/model_garden_mammut.ipynb b/notebooks/community/model_garden/model_garden_mammut.ipynb index ea0a4268e..195a23667 100644 --- a/notebooks/community/model_garden/model_garden_mammut.ipynb +++ b/notebooks/community/model_garden/model_garden_mammut.ipynb @@ -222,6 +222,7 @@ " serving_container_args=[],\n", " location=REGION,\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/mammut\"\n", " )\n", "\n", " print(\"The uploaded model name is: \", upload_job_name)\n", diff --git a/notebooks/community/model_garden/model_garden_movinet_action_recognition.ipynb b/notebooks/community/model_garden/model_garden_movinet_action_recognition.ipynb index d13f2e1fb..c6a49f1d2 100644 --- a/notebooks/community/model_garden/model_garden_movinet_action_recognition.ipynb +++ b/notebooks/community/model_garden/model_garden_movinet_action_recognition.ipynb @@ -726,6 +726,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/tfvision-movinet-var\",\n", ")\n", "\n", "models[\"model_var\"].wait()\n", diff --git a/notebooks/community/model_garden/model_garden_movinet_clip_classification.ipynb b/notebooks/community/model_garden/model_garden_movinet_clip_classification.ipynb index 7061384d9..6b0d6262a 100644 --- a/notebooks/community/model_garden/model_garden_movinet_clip_classification.ipynb +++ b/notebooks/community/model_garden/model_garden_movinet_clip_classification.ipynb @@ -770,6 +770,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/tfvision-movinet-vcn\"\n", ")\n", "\n", "model.wait()\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_autogluon.ipynb b/notebooks/community/model_garden/model_garden_pytorch_autogluon.ipynb index 3b511f8f9..c089e4eeb 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_autogluon.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_autogluon.ipynb @@ -311,6 +311,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/autogluon-ai/models/autogluon\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_bart_large_cnn.ipynb b/notebooks/community/model_garden/model_garden_pytorch_bart_large_cnn.ipynb index 827d2a667..34b3aa71b 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_bart_large_cnn.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_bart_large_cnn.ipynb @@ -205,6 +205,7 @@ " serving_container_predict_route=\"/pred\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/bart-large-cnn\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_biogpt_serve.ipynb b/notebooks/community/model_garden/model_garden_pytorch_biogpt_serve.ipynb index f0ddde073..72259e4b6 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_biogpt_serve.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_biogpt_serve.ipynb @@ -256,6 +256,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/microsoft/models/bio-gpt\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_biomedclip.ipynb b/notebooks/community/model_garden/model_garden_pytorch_biomedclip.ipynb index f952b9c49..9b90cb603 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_biomedclip.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_biomedclip.ipynb @@ -251,6 +251,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/microsoft/models/microsoft-biomedclip\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb b/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb index 4d40f7495..d3fb8e67e 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb @@ -285,6 +285,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/salesforce/models/blip2-opt-2.7-b\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb index 6986acf7a..3a548e9f8 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb @@ -284,6 +284,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/salesforce/models/blip-image-captioning-base\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb b/notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb index b88bf93e0..d6aabde5f 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb @@ -207,6 +207,7 @@ " serving_container_predict_route=\"/pred\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/salesforce/models/blip-vqa-base\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_clip.ipynb b/notebooks/community/model_garden/model_garden_pytorch_clip.ipynb index f49f70a6f..f7bac1a30 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_clip.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_clip.ipynb @@ -284,6 +284,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/openai/models/clip-vit-base-patch32\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb b/notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb index 4a975c1f2..819b3148c 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb @@ -213,6 +213,7 @@ " serving_container_predict_route=\"/predictions/diffusers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/lllyasviel/models/control-net\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_dolly_v2.ipynb b/notebooks/community/model_garden/model_garden_pytorch_dolly_v2.ipynb index 1b4f86f10..6d7b3a50a 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_dolly_v2.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_dolly_v2.ipynb @@ -296,6 +296,7 @@ " serving_container_predict_route=\"/generate\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=env_vars,\n", + " model_garden_source_model_name=\"publishers/databricks/models/dolly-v2\"\n", " )\n", " print(\n", " f\"Deploying {model_name} on {machine_type} with {accelerator_count} {accelerator_type} GPU(s).\"\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_deployment.ipynb index 019ecd9f8..c34f9bb5a 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_deployment.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_deployment.ipynb @@ -198,6 +198,7 @@ " serving_container_predict_route=\"/generate\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/tiiuae/models/falcon-instruct-7b-peft\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_finetuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_finetuning.ipynb index 5e3f29e95..1a5388b20 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_finetuning.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_finetuning.ipynb @@ -216,6 +216,7 @@ " serving_container_predict_route=\"/predictions/peft_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/tiiuae/models/falcon-instruct-7b-peft\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", @@ -265,6 +266,7 @@ " serving_container_ports=[7080],\n", " serving_container_predict_route=\"/generate\",\n", " serving_container_health_route=\"/ping\",\n", + " model_garden_source_model_name=\"publishers/tiiuae/models/falcon-instruct-7b-peft\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_peft.ipynb b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_peft.ipynb index a823f2d6d..72d4b857d 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_peft.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_peft.ipynb @@ -301,6 +301,7 @@ " serving_container_predict_route=\"/predictions/peft_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/tiiuae/models/falcon-instruct-7b-peft\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", @@ -354,6 +355,7 @@ " serving_container_ports=[7080],\n", " serving_container_predict_route=\"/generate\",\n", " serving_container_health_route=\"/ping\",\n", + " model_garden_source_model_name=\"publishers/tiiuae/models/falcon-instruct-7b-peft\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_quantization.ipynb b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_quantization.ipynb index fe633af4c..08f67a500 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_quantization.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_quantization.ipynb @@ -212,6 +212,7 @@ " serving_container_predict_route=\"/generate\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/tiiuae/models/falcon-instruct-7b-peft\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_flux.ipynb b/notebooks/community/model_garden/model_garden_pytorch_flux.ipynb index 6cec02e00..19dcf7935 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_flux.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_flux.ipynb @@ -258,6 +258,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/black-forest-labs/models/flux1-schnell\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_flux_gradio.ipynb b/notebooks/community/model_garden/model_garden_pytorch_flux_gradio.ipynb index 6be14bb46..795b64f56 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_flux_gradio.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_flux_gradio.ipynb @@ -265,6 +265,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/black-forest-labs/models/flux1-schnell\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_imagebind.ipynb b/notebooks/community/model_garden/model_garden_pytorch_imagebind.ipynb index e62e9b48e..cafe1debc 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_imagebind.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_imagebind.ipynb @@ -276,6 +276,7 @@ " serving_container_predict_route=\"/predictions/imagebind_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/meta/models/imagebind\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_instant_id.ipynb b/notebooks/community/model_garden/model_garden_pytorch_instant_id.ipynb index 4e9906db7..b351484f3 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_instant_id.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_instant_id.ipynb @@ -225,6 +225,7 @@ " serving_container_predict_route=\"/predictions/diffusers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/instantx/models/instant-id\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_instant_id_gradio.ipynb b/notebooks/community/model_garden/model_garden_pytorch_instant_id_gradio.ipynb index b0783eb50..bafe51d0c 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_instant_id_gradio.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_instant_id_gradio.ipynb @@ -312,6 +312,7 @@ " serving_container_predict_route=\"/predictions/diffusers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/instantx/models/instant-id\"\n", " )\n", " machine_type = \"g2-standard-8\"\n", " accelerator_type = \"NVIDIA_L4\"\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb b/notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb index 12fcc2986..138bf2e29 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb @@ -208,6 +208,7 @@ " serving_container_predict_route=\"/predictions/diffusers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/timbrooks/models/instruct-pix2pix\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_lama.ipynb b/notebooks/community/model_garden/model_garden_pytorch_lama.ipynb index b4c2a6da8..35e6e7b3c 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_lama.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_lama.ipynb @@ -302,6 +302,7 @@ " serving_container_predict_route=\"/predictions/lama\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/advimman/models/lama\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb b/notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb index e13b363fe..08830bbfe 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb @@ -284,6 +284,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/impira/models/layoutlm-document-qa\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama2_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama2_deployment.ipynb index 4a965c6e7..e27f1d95c 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_llama2_deployment.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_llama2_deployment.ipynb @@ -254,6 +254,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=env_vars,\n", " artifact_uri=model_id,\n", + " model_garden_source_model_name=\"publishers/meta/models/llama2\"\n", " )\n", " print(\n", " f\"Deploying {model_name} on {machine_type} with {accelerator_count} {accelerator_type} GPU(s).\"\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_finetuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_finetuning.ipynb index 182839bf5..0e9823233 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_finetuning.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_finetuning.ipynb @@ -277,6 +277,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=env_vars,\n", " artifact_uri=model_id,\n", + " model_garden_source_model_name=\"publishers/meta/models/llama2\"\n", " )\n", " print(\n", " f\"Deploying {model_name} on {machine_type} with {accelerator_count} {accelerator_type} GPU(s).\"\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_hyperparameter_tuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_hyperparameter_tuning.ipynb index 0eeb9d147..2621d2b09 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_hyperparameter_tuning.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_llama2_peft_hyperparameter_tuning.ipynb @@ -231,6 +231,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=env_vars,\n", " artifact_uri=model_id,\n", + " model_garden_source_model_name=\"publishers/meta/models/llama2\"\n", " )\n", " print(\n", " f\"Deploying {model_name} on {machine_type} with {accelerator_count} {accelerator_type} GPU(s).\"\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_nllb.ipynb b/notebooks/community/model_garden/model_garden_pytorch_nllb.ipynb index bc402a007..a9173d07b 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_nllb.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_nllb.ipynb @@ -239,6 +239,7 @@ " serving_container_predict_route=\"/predictions/transformers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/meta/models/nllb\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_open_clip.ipynb b/notebooks/community/model_garden/model_garden_pytorch_open_clip.ipynb index c40dcdfff..125424c1b 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_open_clip.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_open_clip.ipynb @@ -365,6 +365,7 @@ " serving_container_predict_route=\"/predictions/transformers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/openai/models/openclip\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_openllama_peft.ipynb b/notebooks/community/model_garden/model_garden_pytorch_openllama_peft.ipynb index 95d6258c4..b5b7b5d98 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_openllama_peft.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_openllama_peft.ipynb @@ -304,6 +304,7 @@ " serving_container_predict_route=\"/predictions/peft_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/openlm-research/models/openllama\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", @@ -359,6 +360,7 @@ " serving_container_predict_route=\"/generate\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/openlm-research/models/openllama\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb b/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb index 7d4c8a97d..16b64e67c 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb @@ -239,6 +239,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/google/models/owlvit-base-patch32\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_pic2word.ipynb b/notebooks/community/model_garden/model_garden_pytorch_pic2word.ipynb index 8f2de58b9..f1b8814f2 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_pic2word.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_pic2word.ipynb @@ -282,6 +282,7 @@ " serving_container_predict_route=\"/predictions/pic2word\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/pic2word\"\n", ")\n", "\n", "# Create an endpoint.\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_sam.ipynb b/notebooks/community/model_garden/model_garden_pytorch_sam.ipynb index 6eb6d584a..6024810af 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_sam.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_sam.ipynb @@ -245,6 +245,7 @@ " serving_container_predict_route=\"/predictions/sam_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/meta/models/segment-anything\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_deployment.ipynb index d0df60c24..79d14ae39 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_deployment.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_deployment.ipynb @@ -212,6 +212,7 @@ " serving_container_predict_route=\"/predictions/diffusers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-2-1\",\n", " )\n", " else:\n", " model = aiplatform.Model.upload(\n", @@ -221,6 +222,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-2-1\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_finetuning_dreambooth.ipynb b/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_finetuning_dreambooth.ipynb index a4c8cf600..38f5b9bdb 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_finetuning_dreambooth.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_sd_2_1_finetuning_dreambooth.ipynb @@ -392,6 +392,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-2-1\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_sd_xl_finetuning_dreambooth_lora.ipynb b/notebooks/community/model_garden/model_garden_pytorch_sd_xl_finetuning_dreambooth_lora.ipynb index d30bdbaa5..789b935ec 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_sd_xl_finetuning_dreambooth_lora.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_sd_xl_finetuning_dreambooth_lora.ipynb @@ -440,6 +440,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-xl-base\",\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_custom.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_custom.ipynb index 150eab47b..e60ef5511 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_custom.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_custom.ipynb @@ -234,6 +234,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-xl-base\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb index 2d014874a..35614c9c6 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb @@ -210,6 +210,7 @@ " serving_container_predict_route=\"/predictions/diffusers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/runwayml/models/stable-diffusion-inpainting\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_upscaler.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_upscaler.ipynb index 5413a5e0d..b123bfa10 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_upscaler.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_upscaler.ipynb @@ -224,6 +224,7 @@ " serving_container_predict_route=\"/predictions/diffusers_serving\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-4x-upscaler\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_1_0.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_1_0.ipynb index bef3054fd..21d97f513 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_1_0.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_1_0.ipynb @@ -214,6 +214,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-xl-base\",\n", " )\n", " machine_type = \"g2-standard-8\"\n", " accelerator_type = \"NVIDIA_L4\"\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lcm.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lcm.ipynb index d6d3156f6..f359a34cf 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lcm.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lcm.ipynb @@ -229,6 +229,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-xl-lcm\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lightning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lightning.ipynb index 7efc31717..cb70c9b6f 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lightning.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lightning.ipynb @@ -209,6 +209,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/bytedance/models/stable-diffusion-xl-lightning\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lora.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lora.ipynb index 483e03bb0..6ba54ce50 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lora.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lora.ipynb @@ -226,6 +226,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-xl-base\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_turbo.ipynb b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_turbo.ipynb index 639d72af6..e654dfea5 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_turbo.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_turbo.ipynb @@ -225,6 +225,7 @@ " serving_container_predict_route=\"/predict\",\n", " serving_container_health_route=\"/health\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/stability-ai/models/stable-diffusion-xl-base\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_vicuna.ipynb b/notebooks/community/model_garden/model_garden_pytorch_vicuna.ipynb index 481968067..39b73fe11 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_vicuna.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_vicuna.ipynb @@ -301,6 +301,7 @@ " serving_container_predict_route=\"/generate\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/lmsys/models/lmsys-vicuna-7b\"\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb b/notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb index 4d0add3ba..887649c87 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb @@ -206,6 +206,7 @@ " serving_container_predict_route=\"/pred\",\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/dandelin/models/vilt-b32-finetuned-vqa\",\n", " )\n", "\n", " model.deploy(\n", diff --git a/notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb index f4c8ec83e..b52b3c2f6 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb @@ -284,6 +284,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", + " model_garden_source_model_name=\"publishers/nlpconnect/models/vit-gpt2-image-captioning\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", diff --git a/notebooks/community/model_garden/model_garden_tfvision_image_segmentation.ipynb b/notebooks/community/model_garden/model_garden_tfvision_image_segmentation.ipynb index b86bbd2db..b9e9eeb21 100644 --- a/notebooks/community/model_garden/model_garden_tfvision_image_segmentation.ipynb +++ b/notebooks/community/model_garden/model_garden_tfvision_image_segmentation.ipynb @@ -663,6 +663,7 @@ " serving_container_image_uri=PREDICTION_CONTAINER_URI,\n", " serving_container_args=SERVING_CONTAINER_ARGS,\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/google/models/imagesegmentation-deeplabv3\",\n", ")\n", "\n", "models[\"model_isg\"].wait()\n", diff --git a/notebooks/community/model_garden/model_garden_timesfm_deployment_on_vertex.ipynb b/notebooks/community/model_garden/model_garden_timesfm_deployment_on_vertex.ipynb index afa85b6cc..3351a99df 100644 --- a/notebooks/community/model_garden/model_garden_timesfm_deployment_on_vertex.ipynb +++ b/notebooks/community/model_garden/model_garden_timesfm_deployment_on_vertex.ipynb @@ -351,6 +351,7 @@ " \"TIMESFM_BACKEND\": timesfm_backend,\n", " },\n", " credentials=aiplatform.initializer.global_config.credentials,\n", + " model_garden_source_model_name=\"publishers/google/models/timesfm\",\n", " )\n", " print(\n", " f\"Deploying {model_name_with_time} on {machine_type} with\"\n", diff --git a/notebooks/community/model_garden/model_garden_video_object_tracking_serve.ipynb b/notebooks/community/model_garden/model_garden_video_object_tracking_serve.ipynb index 3785386fc..9b14fe8c8 100644 --- a/notebooks/community/model_garden/model_garden_video_object_tracking_serve.ipynb +++ b/notebooks/community/model_garden/model_garden_video_object_tracking_serve.ipynb @@ -441,6 +441,7 @@ " serving_container_predict_route=SERVE_ROUTE,\n", " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", + " model_garden_source_model_name=\"publishers/ifzhang/models/bytetrack-multi-object-tracking\"\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n",