01.AI/Dense

Yi-1.5 6B

chat
6.06B
Parameters
4K
Context length
8
Benchmarks
4
Quantizations
80K
HF downloads
Architecture
Dense
Released
2024-05-13
Layers
32
KV Heads
4
Head Dim
128
Family
yi
<div align="center"> <picture> <img src="https://raw.githubusercontent.com/01-ai/Yi/main/assets/img/Yi_logo_icon_light.svg" width="150px"> </picture> </div> <p align="center"> <a href="https://github.com/01-ai">🐙 GitHub</a> • <a href="https://discord.gg/hYUwWddeAu">👾 Discord</a> • <a href="https://twitter.com/01ai_yi">🐤 Twitter</a> • <a href="https://github.com/01-ai/Yi-1.5/issues/2">💬 WeChat</a> <br/> <a href="https://arxiv.org/abs/2403.04652">📝 Paper</a> • <a href="https://01-ai.github.io/">💪 Tech Blog</a> • <a href="https://github.com/01-ai/Yi/tree/main?tab=readme-ov-file#faq">🙌 FAQ</a> • <a href="https://github.com/01-ai/Yi/tree/main?tab=readme-ov-file#learning-hub">📗 Learning Hub</a> </p>

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.

<div align="center">
ModelContext LengthPre-trained Tokens
Yi-1.54K, 16K, 32K3.6T
</div>

Models

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.

Quantizations & VRAM

Q4_K_M4.5 bpw
3.9 GB
VRAM required
94%
Quality
Q6_K6.5 bpw
5.4 GB
VRAM required
97%
Quality
Q8_08 bpw
6.5 GB
VRAM required
100%
Quality
FP1616 bpw
12.6 GB
VRAM required
100%
Quality

Benchmarks (8)

IFEval51.5
MATH45.0
HumanEval38.0
BigCodeBench25.6
MMLU-PRO24.4
BBH23.7
MUSR14.0
GPQA6.9

GPUs that can run this model

At Q4_K_M quantization. Sorted by minimum VRAM.

Find the best GPU for Yi-1.5 6B

Build Hardware for Yi-1.5 6B