Downloads · 30 days
14.9K
6% of all-time downloads
Yi-1.5-6B
01-ai/Yi-1.5-6B
Yi-1.5-6B is a text generation model from 01-ai. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
All-time downloads
264K
Public
Parameters
6.1B
12.1 GB on disk
Likes
32
Public
Repo makeup
Click a slice to open those files.
.safetensors12.1 GB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- apache-2.0
- Model type
- llama
- Access
- Public
- Created
- May 11, 2024
- Updated
- Jun 26, 2024
- SHA
- 157a3d77
Try a prompt
Model card
From the Hugging Face model README
<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>
</div>
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">| Model | Context Length | Pre-trained Tokens |
|---|---|---|
| Yi-1.5 | 4K, 16K, 32K | 3.6T |
Models
-
Chat models
<div align="center"></div>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
<div align="center"></div>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.
Yi-1.5-9B-Chat is the top performer among similarly sized open-source models.
-
Base models
Yi-1.5-34B is on par with or excels beyond larger models in some benchmarks.
Yi-1.5-9B is the top performer among similarly sized open-source models.
Quick Start
For getting up and running with Yi-1.5 models quickly, see README.
- Task
- Text Generation
- Library
- transformers
- Type
- llama
- License
- apache-2.0
- Created
- May 11, 2024
- Updated
- Jun 26, 2024