Prediction

Model

01-ai/yi-6b:d302e64f

pwsib73boiljkblsdl4vp6j5zi

Status

Succeeded

Source

Web

Hardware

A40 (Large)

Total duration

0.4s

Created

about 2 years ago

Webhook

–

Input

prompt: Question: I had 3 apples and I ate 1. How many apples do I have left? Answer:
max_new_tokens: 64
temperature: 0.9
top_p: 0.95
top_k: 50
presence_penalty: 1
frequency_penalty: 1
prompt_template: {prompt}

{
  "frequency_penalty": 1,
  "max_new_tokens": 64,
  "presence_penalty": 1,
  "prompt": "Question: I had 3 apples and I ate 1. How many apples do I have left?\nAnswer:",
  "prompt_template": "{prompt}",
  "temperature": 0.9,
  "top_k": 50,
  "top_p": 0.95
}

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=r8_XLZ**********************************

This is your API token. Keep it to yourself.

Import and set up the client:

import Replicate from "replicate";

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

Run 01-ai/yi-6b using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "01-ai/yi-6b:d302e64fad6b4d85d47b3d1ed569b06107504f5717ee1ec12136987bec1e94f1",
  {
    input: {
      frequency_penalty: 1,
      max_new_tokens: 64,
      presence_penalty: 1,
      prompt: "Question: I had 3 apples and I ate 1. How many apples do I have left?\nAnswer:",
      prompt_template: "{prompt}",
      temperature: 0.9,
      top_k: 50,
      top_p: 0.95
    }
  }
);

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=r8_XLZ**********************************

This is your API token. Keep it to yourself.

Import the client:

import replicate

Run 01-ai/yi-6b using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "01-ai/yi-6b:d302e64fad6b4d85d47b3d1ed569b06107504f5717ee1ec12136987bec1e94f1",
    input={
        "frequency_penalty": 1,
        "max_new_tokens": 64,
        "presence_penalty": 1,
        "prompt": "Question: I had 3 apples and I ate 1. How many apples do I have left?\nAnswer:",
        "prompt_template": "{prompt}",
        "temperature": 0.9,
        "top_k": 50,
        "top_p": 0.95
    }
)

# The 01-ai/yi-6b 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/01-ai/yi-6b/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=r8_XLZ**********************************

This is your API token. Keep it to yourself.

Run 01-ai/yi-6b 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": "01-ai/yi-6b:d302e64fad6b4d85d47b3d1ed569b06107504f5717ee1ec12136987bec1e94f1",
    "input": {
      "frequency_penalty": 1,
      "max_new_tokens": 64,
      "presence_penalty": 1,
      "prompt": "Question: I had 3 apples and I ate 1. How many apples do I have left?\\nAnswer:",
      "prompt_template": "{prompt}",
      "temperature": 0.9,
      "top_k": 50,
      "top_p": 0.95
    }
  }' \
  https://api.replicate.com/v1/predictions

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

Output

 After eating one apple, I had 2 apples left.
(H) - (A) = A

Generated in

0.3 seconds

Tweak it Report