Skip to content

Commit d19ff2d

Browse files
fix: pipeline template for LLMaaJ base model inference (#5582)
Co-authored-by: Zhaoqi <52220743+zhaoqizqwang@users.noreply.github.com>
1 parent 487764a commit d19ff2d

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)