AI Agent Hub
Back to models
Yi-1.5-9B-Chat-16K logo

Yi-1.5-9B-Chat-16K

Open Source 01-ai Released 2024-05-13
-- 9.0B params 16.4K context Open Source

About this model

πŸ™ GitHub β€’ πŸ‘Ύ Discord β€’ 🐀 Twitter β€’ πŸ’¬ WeChat
πŸ“ Paper β€’ πŸ’ͺ Tech Blog β€’ πŸ™Œ FAQ β€’ πŸ“— Learning Hub

Intro

Yi-1.5 is an upgraded version of Yi. It is continuously pre-trained on Yi with a high-quality corpus of 500B tokens and fine-tuned on 3M diverse fine-tuning samples.

Compared with Yi, Yi-1.5 delivers stronger performance in coding, math, reasoning, and instruction-following capability, while still maintaining excellent capabilities in language understanding, commonsense reasoning, and reading comprehension.

Model Context Length Pre-trained Tokens
Yi-1.5 4K, 16K, 32K 3.6T

Models

  • Chat models
Name Download
Yi-1.5-34B-Chat β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-34B-Chat-16K β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-9B-Chat β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-9B-Chat-16K β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-6B-Chat β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
  • Base models
Name Download
Yi-1.5-34B β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-34B-32K β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-9B β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-9B-32K β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel
Yi-1.5-6B β€’ πŸ€— Hugging Face β€’ πŸ€– ModelScope β€’ 🟣 wisemodel

Benchmarks

  • Chat models

Yi-1.5-34B-Chat is on par with or excels beyond larger models in most benchmarks.

image/png

Yi-1.5-9B-Chat is the top performer among similarly sized open-source models.

image/png

  • Base models

Yi-1.5-34B is on par with or excels beyond larger models in some benchmarks.

image/png

Yi-1.5-9B is the top performer among similarly sized open-source models.

image/png

Quick Start

For getting up and running with Yi-1.5 models quickly, see README.

Benchmark Scores

ARC
60.8
BBH
51.2
GPQA
30.9
MATH
47.7
MMLU
69.5
GSM8K
84.8
IFEval
35.1
MMLU-Pro
39.9
HumanEval
66.5
GPQA-Diamond
32.8

Technical Specs

  • Parameters: 9.0B
  • Architecture: Transformer
  • Context Window: 16,384 tokens
  • Input Modalities: text

Hardware Requirements

  • VRAM: 18.0 GB
  • Compute: Single NVIDIA GPU with 18GB+ VRAM (e.g. RTX 3090/4090, A10) for BF16 inference