diff --git a/charts/model-engine/templates/service_account.yaml b/charts/model-engine/templates/service_account.yaml index c226505f..63ca6d27 100644 --- a/charts/model-engine/templates/service_account.yaml +++ b/charts/model-engine/templates/service_account.yaml @@ -16,6 +16,11 @@ metadata: {{- if $.Values.azure }} azure.workload.identity/client-id: {{ $.Values.azure.client_id }} {{- end }} + {{- if $.Values.gcp }} + {{- if $.Values.gcp.iam_service_account }} + iam.gke.io/gcp-service-account: {{ $.Values.gcp.iam_service_account }} + {{- end }} + {{- end }} {{- end }} {{- if $.Values.azure }} imagePullSecrets: diff --git a/charts/model-engine/templates/service_account_inference.yaml b/charts/model-engine/templates/service_account_inference.yaml index 712f9df5..4907a05d 100644 --- a/charts/model-engine/templates/service_account_inference.yaml +++ b/charts/model-engine/templates/service_account_inference.yaml @@ -20,6 +20,13 @@ metadata: azure.workload.identity/client-id: {{ $.Values.azure.client_id }} {{- end }} {{- end }} + {{- if $.Values.gcp }} + {{- if $.Values.gcp.inference_service_account }} + iam.gke.io/gcp-service-account: {{ $.Values.gcp.inference_service_account }} + {{- else if $.Values.gcp.iam_service_account }} + iam.gke.io/gcp-service-account: {{ $.Values.gcp.iam_service_account }} + {{- end }} + {{- end }} {{- end }} {{- if $.Values.azure }} imagePullSecrets: