From 62932161288a884937d39561787c1d75be88eec2 Mon Sep 17 00:00:00 2001
From: Nicolas Jourdan <njourdan@amazon.de>
Date: Thu, 26 Feb 2026 10:12:42 +0000
Subject: [PATCH] fix: pipeline template for LLMaaJ base model inference

---
 .../src/sagemaker/train/evaluate/pipeline_templates.py         | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/sagemaker-train/src/sagemaker/train/evaluate/pipeline_templates.py b/sagemaker-train/src/sagemaker/train/evaluate/pipeline_templates.py
index 79eb565fa6..b14145ff31 100644
--- a/sagemaker-train/src/sagemaker/train/evaluate/pipeline_templates.py
+++ b/sagemaker-train/src/sagemaker/train/evaluate/pipeline_templates.py
@@ -1028,6 +1028,9 @@
                 {% if kms_key_id %},
                 "KmsKeyId": "{{ kms_key_id }}"
                 {% endif %}
+                },
+                "ModelPackageConfig": {
+                    "ModelPackageGroupArn": "{{ model_package_group_arn }}"
                 }{% if dataset_uri %},
                 "InputDataConfig": [
                     {