Using Yi 34B 200K on Replicate API

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

ServerlessOpen Source

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

Last refreshed 2026-09-22. Next refresh: weekly.

Quick Start

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

Code Examples

Install
pip install replicate
API key
REPLICATE_API_TOKEN
Model ID
01-ai/yi-34b-200k

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-200k format: "owner/model-name" (latest version) or "owner/model-name:version-hash"
output = replicate.run(
    "01-ai/yi-34b-200k",
    input={"prompt": "Hello"}
)
# Output is a list or generator depending on the model
print("".join(output))

Pricing on Replicate API

TypePrice (per 1M)
Input tokens$0.20
Output tokens$1.00

Capabilities

No model capability flags are currently sourced.

About Yi 34B 200K

The Yi 34B 200K is a sophisticated large language model by 01.AI that excels in varied NLP tasks, featuring an impressive 34 billion parameters and a 200,000-token context window to handle extensive text inputs. Built on a Transformer architecture, it differentiates itself from models like Llama by employing unique training methods such as Grouped-Query Attention and RoPE with adjusted base frequency. It showcases strengths in language comprehension, commonsense reasoning, and bilingual support for English and Chinese. Despite its advanced capabilities, it shares common LLM limitations like hallucination and non-determinism.

Model Specs

Released2023-11-02
Parameters34B
Context200k
ArchitectureDecoder Only
Knowledge cutoff2024-03

More Models on Replicate API

Provider

Replicate API

Replicate

San Francisco, California, United States