Amazon SageMaker JumpStart

Setup

By integrating your Amazon SageMaker Jumpstart into Datasaur, you would be able to spawn a text-generation model on explore tab, retrieve the text-generation model on my models tab and use the models on LLM Labs Playground.

Connect to Amazon SageMaker Jumpstart

1. Select Amazon SageMaker JumpStart.

Manage Providers

2. A new window will pop up. External ID is generated for each time the screen appear. It will be used when create Role ARN later. To prevent the system from regenerating a different External ID, do not close the popup during the setup, otherwise you will need to set up the role ARN with a different External ID.

Amazon SageMaker JumpStart

3. You can start by selecting the region attribute. We'll get back to this window later. Let's leave it for now.

Create a Role ARN for Amazon Sagemaker JumpStart

Create a role which will be assumed by Datasaur to access your Amazon SageMaker Jumpstart.

1. Open the IAM console at https://console.aws.amazon.com/iam/.

2. Choose Roles and then choose Create role.

3. Choose Custom trust policy in the trusted entity type section.

Custom Trust Policy

4. Paste the following configurations in custom trust policy. Don't forget to replace the ExternalID.

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Principal": {
                "AWS": "682361690817"
            },
            "Action": "sts:AssumeRole",
            "Condition": {
                "StringEquals": {
                    "sts:ExternalId": "<externalID>"
                }
            }
        },
        {
            "Effect": "Allow",
            "Principal": {
                "Service": "sagemaker.amazonaws.com"
            },
            "Action": "sts:AssumeRole"
        }
    ]
}

After that, click on Next.

5. In the Add permissions section, pick the policy AmazonSageMakerFullAccess. Then, click on Next.

Add Permissions

6. Enter a Role name and a Description (optional).

7. Choose Create role.

8. After that, back on the Roles page, click on your newly created role.

9. Copy the Role ARN from the page and paste it in Amazon SageMaker Jumpstart manage provider dialog. And click Add Credentials to connect.

Disconnect

1. Click See Details for Amazon SageMaker Jumpstart provider on Manage Providers Dialog.

2. Click Disconnect. All Amazon SageMaker Jumpstart models will be removed from My Models tab, and it will be not accessible on LLM Labs Playground.