AI Agent Hub
Back to models
Yi-1.5-6B logo

Yi-1.5-6B

Open Source 01-ai Released 2024-05-13
-- 6.0B params 4.1K 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

BBH
59.0
MATH
28.4
MBPP
56.8
MMLU
63.5
GSM8K
62.2
AGIEval
56.0
HumanEval
36.5

Technical Specs

  • Parameters: 6.0B
  • Architecture: Transformer
  • Context Window: 4,096 tokens
  • Input Modalities: text

Hardware Requirements

  • VRAM: 12.0 GB
  • Compute: Single NVIDIA GPU with 16GB VRAM for FP16 inference (12GB model weights)