Yi

Stale
GitHub Jupyter Notebook Apache-2.0

Description

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

💡 Pick an OSS LLM as an OpenAI alternative for Chinese apps.
💡 Long-document summarization and QA.
💡 Provide LLM backend for local agents.

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', []))

Related Projects