Service account error in Vertex AI pipeline while uploading the fine-tuned model to bucket (Collab)

Google Cloud Build with AI Custom ML & MLOps
,
1

Hi,

I am trying to fine-tune foundation model (llama2-7B) through the Vertex AI model garden fine-tuning. And I am using Collab to perform the PEFT tuning.

As per the given instruction in the notebook, I have created a new service account with roles

Vertex AI User and Storage Object Admin.

# The service account looks like:
# '@.iam.gserviceaccount.com'
# Please go to https://cloud.google.com/iam/docs/service-accounts-create#iam-service-accounts-create-console
# and create service account with `Vertex AI User` and `Storage Object Admin` roles.
# The service account for deploying fine tuned model.

Pipeline gets executed and the fine-tuning step has been successfully completed.

However in the upload-model , I get the below error.

ValueError: Failed to create the resource. Error: {'code': 400, 'message': 'You do not have permission to act as service_account: xyz@xyz-stag.iam.gserviceaccount.com. (or it may not exist).', 'status': 'INVALID_ARGUMENT'}

Can you please help me identifying the issue here and help me fix the issue?

Thank you,

KK

2

After trial and error, I was able to fix the issue.

Along with Vertex AI User and Storage Object Admin, we should also add Service Account User as one of the role.

Thank you,

KK