Deploy AI21 Labs Language Models on AWS SageMaker

Examples for using Jurassic-2 models through Amazon SageMaker.

Installation

You can install the package using pip:

pip install -U "ai21[AWS]"

And then import the package:

import ai21

Quickstart

In order to use this library, you must configure your AWS credentials.

All set up? Make your first interaction with Jurassic-2:

response = ai21.Completion.execute(destination=ai21.SageMakerDestination("j2-ultra"),
                                   prompt="To be, or",
                                   maxTokens=4,
                                   temperature=0)

print(response['completions'][0]['data']['text'])
# not to be: that is the question