Yi
StaleDescription
01.AI's open-source LLM series with balanced Chinese and English capabilities.
Key Features
- Bilingual SOTA — Top scores on Chinese + English benchmarks
- Multiple sizes — 6B, 9B, 34B series
- Commercial-friendly — Apache 2.0 license
- Native 200K context — Strong long-doc performance
- Tool calling — OpenAI-compatible tool calling
Use Cases
Tags
Categories
Quick Start
# Load via transformers
pip install transformers torch
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained('01-ai/Yi-1.5-9B-Chat')
print(model.chat('Hello, please introduce yourself', []))