Back to Open Interpreter

AWS Sagemaker

docs/language-models/hosted-models/aws-sagemaker.mdx

0.4.22.8 KB
Original Source

To use Open Interpreter with a model from AWS Sagemaker, set the model flag:

<CodeGroup>
bash
interpreter --model sagemaker/<model-name>
python
# Sagemaker requires boto3 to be installed on your machine:
!pip install boto3

from interpreter import interpreter

interpreter.llm.model = "sagemaker/<model-name>"
interpreter.chat()
</CodeGroup>

Supported Models

We support the following completion models from AWS Sagemaker:

  • Meta Llama 2 7B
  • Meta Llama 2 7B (Chat/Fine-tuned)
  • Meta Llama 2 13B
  • Meta Llama 2 13B (Chat/Fine-tuned)
  • Meta Llama 2 70B
  • Meta Llama 2 70B (Chat/Fine-tuned)
  • Your Custom Huggingface Model
<CodeGroup>
bash
interpreter --model sagemaker/jumpstart-dft-meta-textgeneration-llama-2-7b
interpreter --model sagemaker/jumpstart-dft-meta-textgeneration-llama-2-7b-f
interpreter --model sagemaker/jumpstart-dft-meta-textgeneration-llama-2-13b
interpreter --model sagemaker/jumpstart-dft-meta-textgeneration-llama-2-13b-f
interpreter --model sagemaker/jumpstart-dft-meta-textgeneration-llama-2-70b
interpreter --model sagemaker/jumpstart-dft-meta-textgeneration-llama-2-70b-b-f
interpreter --model sagemaker/<your-huggingface-deployment-name>
python
interpreter.llm.model = "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-7b"
interpreter.llm.model = "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-7b-f"
interpreter.llm.model = "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-13b"
interpreter.llm.model = "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-13b-f"
interpreter.llm.model = "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-70b"
interpreter.llm.model = "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-70b-b-f"
interpreter.llm.model = "sagemaker/<your-huggingface-deployment-name>"
</CodeGroup>

Required Environment Variables

Set the following environment variables (click here to learn how) to use these models.

Environment VariableDescriptionWhere to Find
AWS_ACCESS_KEY_IDThe API access key for your AWS account.AWS Account Overview -> Security Credentials
AWS_SECRET_ACCESS_KEYThe API secret access key for your AWS account.AWS Account Overview -> Security Credentials
AWS_REGION_NAMEThe AWS region you want to useAWS Account Overview -> Navigation bar -> Region