Skip to content

Commit 6293216

Browse files
committed
fix: pipeline template for LLMaaJ base model inference
1 parent 723455f commit 6293216

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

sagemaker-train/src/sagemaker/train/evaluate/pipeline_templates.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1028,6 +1028,9 @@
10281028
{% if kms_key_id %},
10291029
"KmsKeyId": "{{ kms_key_id }}"
10301030
{% endif %}
1031+
},
1032+
"ModelPackageConfig": {
1033+
"ModelPackageGroupArn": "{{ model_package_group_arn }}"
10311034
}{% if dataset_uri %},
10321035
"InputDataConfig": [
10331036
{

0 commit comments

Comments
 (0)