pku-yuangroup/llava-cot:40c17578 – Run with an API on Replicate

Version

You're looking at a specific version of this model. Jump to the model overview.

pku-yuangroup /llava-cot:40c17578

Playground API

Input

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

npx create-replicate --model=pku-yuangroup/llava-cot

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

const output = await replicate.run(
  "pku-yuangroup/llava-cot:40c17578a6b98679fc08e1a92fcef1b8000f857bf300a381091a13a67c28754d",
  {
    input: {
      top_p: 0.95,
      prompt: "If I had to write a haiku for this one, it would be: ",
      temperature: 0.9,
      max_new_tokens: 1024
    }
  }
);
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 pku-yuangroup/llava-cot using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "pku-yuangroup/llava-cot:40c17578a6b98679fc08e1a92fcef1b8000f857bf300a381091a13a67c28754d",
    input={
        "top_p": 0.95,
        "prompt": "If I had to write a haiku for this one, it would be: ",
        "temperature": 0.9,
        "max_new_tokens": 1024
    }
)

# The pku-yuangroup/llava-cot 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/pku-yuangroup/llava-cot/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 pku-yuangroup/llava-cot 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": "40c17578a6b98679fc08e1a92fcef1b8000f857bf300a381091a13a67c28754d",
    "input": {
      "top_p": 0.95,
      "prompt": "If I had to write a haiku for this one, it would be: ",
      "temperature": 0.9,
      "max_new_tokens": 1024
    }
  }' \
  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/pku-yuangroup/llava-cot@sha256:40c17578a6b98679fc08e1a92fcef1b8000f857bf300a381091a13a67c28754d \
  -i 'top_p=0.95' \
  -i 'prompt="If I had to write a haiku for this one, it would be: "' \
  -i 'temperature=0.9' \
  -i 'max_new_tokens=1024'

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/pku-yuangroup/llava-cot@sha256:40c17578a6b98679fc08e1a92fcef1b8000f857bf300a381091a13a67c28754d
curl -s -X POST \
  -H "Content-Type: application/json" \
  -d $'{
    "input": {
      "top_p": 0.95,
      "prompt": "If I had to write a haiku for this one, it would be: ",
      "temperature": 0.9,
      "max_new_tokens": 1024
    }
  }' \
  http://localhost:5000/predictions

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

Output

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