Yi-6B
Parameters
6B
Context Length
4K
Modality
Text
Architecture
Dense
License
Apache 2.0
Release Date
2 Nov 2023
Knowledge Cutoff
Jun 2023
API Pricing (per 1M)
Self-hosted only
System Requirements
VRAM requirements for different quantization methods and context sizes
1,024 tokens
Consumer
1x RTX 4090
24GB VRAM
Datacenter
1x NVIDIA A100
80GB VRAM
Apple Silicon
1x Apple M3 Max
128GB VRAM
4,096 tokens
Consumer
1x RTX 4090
24GB VRAM
Datacenter
1x NVIDIA A100
80GB VRAM
Apple Silicon
1x Apple M3 Max
128GB VRAM
Architecture Diagram
Evaluation Benchmarks
No evaluation benchmarks for Yi-6B available.
Rankings
Overall Rank
-
Coding Rank
-
About Yi-6B
Yi-6B is a compact 6B parameter bilingual model developed by 01.AI for fast conversational AI, reasoning, and multilingual text processing. Released under Apache 2.0, it features Grouped-Query Attention across an extensible 200K context window.
Technical Specifications
Attention
Attention Structure
Multi-Head Attention
Attention Heads
32
Key-Value Heads
4
Attention Head Dimension
-
Position Embedding
Absolute Position Embedding
RoPE Theta
5,000,000
Sliding Window Attention
No
Sliding Window Size
-
Sliding Window Ratio
-
Linear Attention
-
Linear Attention Ratio
-
Normalization
RMS Normalization
Activation Function
SwigLU
Dimensions
Hidden Dimension Size
4,096
Number of Layers
32
FFN Intermediate Size (Dense)
11,008
Multi-Token Prediction Heads
-
Tokenizer
Vocabulary Size
64,000