Using Yi 34B on Replicate API

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

ServerlessOpen Source

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

Last refreshed 2026-06-01. 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-34b — see the documentation for request format.
3
You'll be billed $0.20/1M input, $1.00/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-34b

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-34b format: "owner/model-name" (latest version) or "owner/model-name:version-hash"
output = replicate.run(
    "01-ai/yi-34b",
    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.20
Output tokens	$1.00

Capabilities

Structured Outputs

About Yi 34B

Yi 34B is 01.AI's Yi (2023/11) model. It offers a 200K-token context window and scores 43 on MMLU PRO.

Full model details →

Model Specs

Released2023-11-02

Parameters34B

Context200k

ArchitectureDecoder Only

Knowledge cutoff2024-03

Replicate

San Francisco, California, United States