nateraw/codellama-13b | Run with an API on Replicate

Input

Run this model in Node.js with one line of code:

npx create-replicate --model=nateraw/codellama-13b

or set up a project from scratch

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import and set up the client:

import Replicate from "replicate";

const replicate = new Replicate({
  auth: process.env.REPLICATE_API_TOKEN,
});

Run nateraw/codellama-13b using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "nateraw/codellama-13b:223d3018df5f16b50a936ae343577160b154a3cb7622bcfcfb4cc23e2b13122d",
  {
    input: {
      top_k: 50,
      top_p: 0.95,
      message: "def fibonacci(n):\n",
      temperature: 0.8,
      max_new_tokens: 200
    }
  }
);

console.log(output);

To learn more, take a look at the guide on getting started with Node.js.

Install Replicate’s Python client library:

pip install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import the client:

import replicate

Run nateraw/codellama-13b using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "nateraw/codellama-13b:223d3018df5f16b50a936ae343577160b154a3cb7622bcfcfb4cc23e2b13122d",
    input={
        "top_k": 50,
        "top_p": 0.95,
        "message": "def fibonacci(n):\n",
        "temperature": 0.8,
        "max_new_tokens": 200
    }
)

# The nateraw/codellama-13b model can stream output as it's running.
# The predict method returns an iterator, and you can iterate over that output.
for item in output:
    # https://replicate.com/nateraw/codellama-13b/api#output-schema
    print(item, end="")

To learn more, take a look at the guide on getting started with Python.

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run nateraw/codellama-13b using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

curl -s -X POST \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -H "Prefer: wait" \
  -d $'{
    "version": "nateraw/codellama-13b:223d3018df5f16b50a936ae343577160b154a3cb7622bcfcfb4cc23e2b13122d",
    "input": {
      "top_k": 50,
      "top_p": 0.95,
      "message": "def fibonacci(n):\\n",
      "temperature": 0.8,
      "max_new_tokens": 200
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

Output

if n == 0: return 0 elif n == 1: return 1 else: return fibonacci(n - 1) + fibonacci(n - 2) def test_fibonacci(): assert fibonacci(0) == 0 assert fibonacci(1) == 1 assert fibonacci(2) == 1 assert fibonacci(3) == 2 assert fibonacci(4) == 3 assert fibonacci(5) == 5 assert fibonacci(6) == 8 assert fibonacci(7) == 13 assert fibonacci(8) == 21 assert fibonacci(9) == 34 # 3. def is_prime(n

{
  "completed_at": "2023-09-28T08:45:35.337078Z",
  "created_at": "2023-09-28T08:45:09.217969Z",
  "data_removed": false,
  "error": null,
  "id": "kp673o3buin4ddecw4ozl6msku",
  "input": {
    "top_k": 50,
    "top_p": 0.95,
    "message": "def fibonacci(n):\n",
    "temperature": 0.8,
    "max_new_tokens": 200
  },
  "logs": "Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.",
  "metrics": {
    "predict_time": 26.115524,
    "total_time": 26.119109
  },
  "output": [
    "  ",
    " ",
    "if ",
    "n ",
    "== ",
    "",
    "",
    "0:\n",
    "      ",
    " ",
    "return ",
    "",
    "0\n",
    "  ",
    " ",
    "elif ",
    "n ",
    "== ",
    "",
    "",
    "1:\n",
    "      ",
    " ",
    "return ",
    "",
    "1\n",
    "  ",
    " ",
    "",
    "else:\n",
    "      ",
    " ",
    "return ",
    "",
    "",
    "",
    "",
    "fibonacci(n ",
    "- ",
    "",
    "",
    "1) ",
    "+ ",
    "",
    "",
    "",
    "",
    "fibonacci(n ",
    "- ",
    "",
    "",
    "2)\n",
    "\n",
    "\n",
    "",
    "def ",
    "",
    "",
    "",
    "",
    "",
    "",
    "test_fibonacci():\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(0) ",
    "== ",
    "",
    "0\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(1) ",
    "== ",
    "",
    "1\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(2) ",
    "== ",
    "",
    "1\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(3) ",
    "== ",
    "",
    "2\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(4) ",
    "== ",
    "",
    "3\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(5) ",
    "== ",
    "",
    "5\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(6) ",
    "== ",
    "",
    "8\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(7) ",
    "== ",
    "",
    "",
    "13\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(8) ",
    "== ",
    "",
    "",
    "21\n",
    "  ",
    " ",
    "assert ",
    "",
    "",
    "",
    "",
    "",
    "fibonacci(9) ",
    "== ",
    "",
    "",
    "34\n",
    "\n",
    "\n",
    "",
    "# ",
    "",
    "",
    "3.\n",
    "\n",
    "",
    "def ",
    "",
    "",
    "",
    "",
    "is_prime(n"
  ],
  "started_at": "2023-09-28T08:45:09.221554Z",
  "status": "succeeded",
  "urls": {
    "get": "https://api.replicate.com/v1/predictions/kp673o3buin4ddecw4ozl6msku",
    "cancel": "https://api.replicate.com/v1/predictions/kp673o3buin4ddecw4ozl6msku/cancel"
  },
  "version": "c05078a2947e6b8ea8abab976f2ad00961573791c2a4f26dcd40dc046ad2f6a6"
}

Generated in

26.1 seconds