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

Input

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

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

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-7b using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "nateraw/codellama-7b:21615a9eaef0a263533559284e2dad43086551ea0edb13c9f86e1090c068c773",
  {
    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-7b using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

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

# The nateraw/codellama-7b 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-7b/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-7b 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-7b:21615a9eaef0a263533559284e2dad43086551ea0edb13c9f86e1090c068c773",
    "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 isPrime(x): if x < 2: return False elif x == 2: return True else: for i in range(2, x): if x % i == 0: return False return True def isPerfect(n): s = 0 for i in range(1, n): if n % i == 0: s += i return n == s def factorial(n): f = 1 for i in range(1, n+1): f *= i return f

{
  "completed_at": "2023-09-28T08:42:13.646334Z",
  "created_at": "2023-09-28T08:42:02.126237Z",
  "data_removed": false,
  "error": null,
  "id": "243zcedbgiaf277shipvrfs72a",
  "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": 11.544273,
    "total_time": 11.520097
  },
  "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",
    "",
    "def ",
    "",
    "",
    "",
    "",
    "",
    "isPrime(x):\n",
    "  ",
    " ",
    "if ",
    "x ",
    "< ",
    "",
    "",
    "2:\n",
    "      ",
    " ",
    "return ",
    "False\n",
    "  ",
    " ",
    "elif ",
    "x ",
    "== ",
    "",
    "",
    "2:\n",
    "      ",
    " ",
    "return ",
    "True\n",
    "  ",
    " ",
    "",
    "else:\n",
    "      ",
    " ",
    "for ",
    "i ",
    "in ",
    "",
    "",
    "",
    "range(2, ",
    "",
    "x):\n",
    "          ",
    " ",
    "if ",
    "x ",
    "% ",
    "i ",
    "== ",
    "",
    "",
    "0:\n",
    "              ",
    " ",
    "return ",
    "False\n",
    "  ",
    " ",
    "return ",
    "True\n",
    "\n",
    "",
    "def ",
    "",
    "",
    "",
    "",
    "",
    "isPerfect(n):\n",
    "  ",
    " ",
    "s ",
    "= ",
    "",
    "0\n",
    "  ",
    " ",
    "for ",
    "i ",
    "in ",
    "",
    "",
    "",
    "range(1, ",
    "",
    "n):\n",
    "      ",
    " ",
    "if ",
    "n ",
    "% ",
    "i ",
    "== ",
    "",
    "",
    "0:\n",
    "          ",
    " ",
    "s ",
    "+= ",
    "i\n",
    "  ",
    " ",
    "return ",
    "n ",
    "== ",
    "s\n",
    "\n",
    "",
    "def ",
    "",
    "",
    "",
    "",
    "factorial(n):\n",
    "  ",
    " ",
    "f ",
    "= ",
    "",
    "1\n",
    "  ",
    " ",
    "for ",
    "i ",
    "in ",
    "",
    "",
    "",
    "range(1, ",
    "",
    "",
    "",
    "n+1):\n",
    "      ",
    " ",
    "f ",
    "",
    "*= ",
    "i\n",
    "  ",
    " ",
    "return ",
    "f\n",
    ""
  ],
  "started_at": "2023-09-28T08:42:02.102061Z",
  "status": "succeeded",
  "urls": {
    "get": "https://api.replicate.com/v1/predictions/243zcedbgiaf277shipvrfs72a",
    "cancel": "https://api.replicate.com/v1/predictions/243zcedbgiaf277shipvrfs72a/cancel"
  },
  "version": "21615a9eaef0a263533559284e2dad43086551ea0edb13c9f86e1090c068c773"
}

Generated in

11.6 seconds