Yi 6B on Replicate API

Yi (2023/11) · 01.AI

ServerlessOpen Source

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

Why use Yi 6B on Replicate API?

Replicate API offers Yi 6B with pay-as-you-go pricing at $0.05/1M input tokens. Replicate is a cloud-based platform that enables users to run machine learning models easily and efficiently.

Compare Yi 6B across 3 providers to find the best fit for your use case
Input / 1M
$0.050
Output / 1M
$0.25
Cache
Not sourced
Batch
Not sourced

Setup recipe

Python + curl
Install
pip install replicate
Auth
export REPLICATE_API_TOKEN=...
Call
import replicate
output = replicate.run(
    "01-ai/yi-6b",
    input={"prompt": "Hello"}
Model ID
01-ai/yi-6b

Request example

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))

Gotchas

  • Use provider model ID "01-ai/yi-6b", not the LLMReference slug "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.
  • The examples expect REPLICATE_API_TOKEN; rename it only if your application config maps the new variable.

Compare Yi 6B Across Providers

ProviderInput (per 1M)Output (per 1M)
Alibaba Cloud PAI-EAS——
Fireworks AI$0.20$0.20
Replicate API$0.05$0.25

Pricing

TypePrice (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.

Get Started

Model Specs

Released2023-11-02
Parameters6B
Context200k
ArchitectureDecoder Only
Knowledge cutoff2023-06

Related Models on Replicate API