Using Yi 6B on Replicate API

Implementation guide · Yi (2023/11) · 01.AI

ServerlessOpen Source

Replicate API exposes Yi 6B through model ID 01-ai/yi-6b. Use the setup steps, sourced pricing, capabilities, and official provider links below to validate this route before deployment.

Last refreshed 2026-05-19. Next refresh: weekly.

Quick Start

1
Create an account at Replicate API and generate an API key.
2
Use the Replicate API SDK or REST API to call 01-ai/yi-6b — see the documentation for request format.
3
You'll be billed $0.05/1M input, $0.25/1M output tokens. See full pricing.

API Portal Documentation Pricing Model Card

Code Examples

Install

pip install replicate

API key

REPLICATE_API_TOKEN

Model ID

01-ai/yi-6b

Replicate uses "owner/model-name" format (e.g. "meta/meta-llama-3-8b-instruct") for the latest version, or "owner/model-name:version-sha" to pin to a specific version. The REST endpoint splits owner and model-name into the path: /v1/models/{owner}/{model-name}/predictions.

import replicate

# reads REPLICATE_API_TOKEN from env
# 01-ai/yi-6b format: "owner/model-name" (latest version) or "owner/model-name:version-hash"
output = replicate.run(
    "01-ai/yi-6b",
    input={"prompt": "Hello"}
)
# Output is a list or generator depending on the model
print("".join(output))

Pricing on Replicate API

Type	Price (per 1M)
Input tokens	$0.05
Output tokens	$0.25

Capabilities

No model capability flags are currently sourced.

About Yi 6B

Yi 6B is 01.AI's Yi (2023/11) model. It offers a 200K-token context window.

Full model details →

Model Specs

Released2023-11-02

Parameters6B

Context200k

ArchitectureDecoder Only

Knowledge cutoff2023-06

Replicate

San Francisco, California, United States