hamelsmu/axolotl – Run with an API on Replicate

hamelsmu / axolotl

Cold

Public
0 runs
A100 (80GB)
GitHub

Run with an API

Playground API Examples Train README Versions

Input

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

npx create-replicate --model=hamelsmu/axolotl

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

const output = await replicate.run(
  "hamelsmu/axolotl:2fd11c56a88a369d5c517ff5d6867bafe101172548641ac2651e6231f72efd73",
  {
    input: {
      top_k: 50,
      top_p: 0.95,
      do_sample: true,
      temperature: 0.7,
      max_new_tokens: 512,
      prompt_template: "### System:\nBelow is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\nTranslate the input from English to Hinglish\n\n### Input:\n{prompt}\n\n### Response:\n "
    }
  }
);
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 hamelsmu/axolotl using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "hamelsmu/axolotl:2fd11c56a88a369d5c517ff5d6867bafe101172548641ac2651e6231f72efd73",
    input={
        "top_k": 50,
        "top_p": 0.95,
        "do_sample": True,
        "temperature": 0.7,
        "max_new_tokens": 512,
        "prompt_template": "### System:\nBelow is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\nTranslate the input from English to Hinglish\n\n### Input:\n{prompt}\n\n### Response:\n "
    }
)

# The hamelsmu/axolotl 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/hamelsmu/axolotl/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 hamelsmu/axolotl 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": "2fd11c56a88a369d5c517ff5d6867bafe101172548641ac2651e6231f72efd73",
    "input": {
      "top_k": 50,
      "top_p": 0.95,
      "do_sample": true,
      "temperature": 0.7,
      "max_new_tokens": 512,
      "prompt_template": "### System:\\nBelow is an instruction that describes a task. Write a response that appropriately completes the request.\\n\\n### Instruction:\\nTranslate the input from English to Hinglish\\n\\n### Input:\\n{prompt}\\n\\n### Response:\\n "
    }
  }' \
  https://api.replicate.com/v1/predictions

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

You can run this model locally using Cog. First, install Cog:

brew install cog

If you don’t have Homebrew, there are other installation options available.

Run this to download the model and run it in your local environment:

cog predict r8.im/hamelsmu/axolotl@sha256:2fd11c56a88a369d5c517ff5d6867bafe101172548641ac2651e6231f72efd73 \
  -i 'top_k=50' \
  -i 'top_p=0.95' \
  -i 'do_sample=true' \
  -i 'temperature=0.7' \
  -i 'max_new_tokens=512' \
  -i $'prompt_template="### System:\\nBelow is an instruction that describes a task. Write a response that appropriately completes the request.\\n\\n### Instruction:\\nTranslate the input from English to Hinglish\\n\\n### Input:\\n{prompt}\\n\\n### Response:\\n "'

To learn more, take a look at the Cog documentation.

Run this to download the model and run it in your local environment:

docker run -d -p 5000:5000 --gpus=all r8.im/hamelsmu/axolotl@sha256:2fd11c56a88a369d5c517ff5d6867bafe101172548641ac2651e6231f72efd73
curl -s -X POST \
  -H "Content-Type: application/json" \
  -d $'{
    "input": {
      "top_k": 50,
      "top_p": 0.95,
      "do_sample": true,
      "temperature": 0.7,
      "max_new_tokens": 512,
      "prompt_template": "### System:\\nBelow is an instruction that describes a task. Write a response that appropriately completes the request.\\n\\n### Instruction:\\nTranslate the input from English to Hinglish\\n\\n### Input:\\n{prompt}\\n\\n### Response:\\n "
    }
  }' \
  http://localhost:5000/predictions

To learn more, take a look at the Cog documentation.

Output

No output yet! Press "Submit" to start a prediction.

Run time and cost

This model runs on Nvidia A100 (80GB) GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

See how this was deployed by looking at the README of replicate/axolotl-trainer