DeepSeek-V2
DeepSeek · DeepSeek V2 · open weights
- MLA
- RoPE
- Partial RoPE
- Pre-norm
- MoE
- Shared expert
- Dense first layers
Facts and where they come from
| Released | 2024-04 | config.jsonconfig.jsonHugging Face repository creation date (api.createdAt) |
|---|---|---|
| Licence | other | config.jsonconfig.jsonREADME metadata: license |
| Total parameters | 236B | labmodel cardREADME: 236B total parameters, of which 21B are activated |
| Active parameters | 21B | labmodel cardREADME: 21B activated |
| Context length | 160K tokens | config.jsonconfig.jsonmax_position_embeddings |
| Norm placement | pre | codemodelling codetransformers 5.18.0 / repo modelling code for deepseek_v2: input_layernorm before attention, post_attention_layernorm before the MLP |
| Norm type | RMSNorm | codemodelling codetransformers 5.18.0 / repo modelling code for deepseek_v2: input_layernorm before attention, post_attention_layernorm before the MLP |
| QK-norm | no | codemodelling codeno q/k normalisation in the attention block (MLA normalises its latent vectors, which is not QK-norm) |
| Positional encoding | RoPE on 33.3% of each head | config.jsonconfig.jsonqk_rope_head_dim / (qk_nope_head_dim + qk_rope_head_dim): decoupled RoPE |
| Parallel attention and MLP | no | codemodelling codetransformers 5.18.0 / repo modelling code for deepseek_v2: input_layernorm before attention, post_attention_layernorm before the MLP |
Architecture, drawn from the data
MLA 128h, latent 512. Each column is one layer: its token mixer above, its feed-forward block below. Paler columns reuse another layer’s keys and values.
Modelled costs
From the cost model, batch size 1. Totals the lab states are in the table above; differences come from rounding, from what a lab counts, or from parts the model does not describe (listed on the about page).
| Parameters (modelled) | 236B |
|---|---|
| Active per token (modelled) | 21.4B |
| Without embeddings and output head | 235B total, 20.3B active |
| Published weights (Hugging Face count) | 236B |
| KV cache per token, BF16 (layers that grow with context) | 67.5 KiB |
| KV cache + state at 160K tokens, BF16 | 10.5 GiB |
| Decode FLOPs per token at 4K context | 61.8 GFLOP |
| Prefill FLOPs for a 4K prompt | 208 TFLOP |
KV cache against context
Every architecture field
| Field | Value | Source |
|---|---|---|
| d_model | 5,120 | config.jsonconfig.jsonhidden_size |
| vocab | 102,400 | config.jsonconfig.jsonvocab_size |
| tied_embeddings | false | config.jsonconfig.jsontie_word_embeddings |
| mixers.mla.type | mla | codemodelling codemulti-head latent attention |
| mixers.mla.heads | 128 | config.jsonconfig.jsonnum_attention_heads |
| mixers.mla.q_lora_rank | 1,536 | config.jsonconfig.jsonq_lora_rank |
| mixers.mla.kv_lora_rank | 512 | config.jsonconfig.jsonkv_lora_rank |
| mixers.mla.qk_nope | 128 | config.jsonconfig.jsonqk_nope_head_dim |
| mixers.mla.qk_rope | 64 | config.jsonconfig.jsonqk_rope_head_dim |
| mixers.mla.v_head_dim | 128 | config.jsonconfig.jsonv_head_dim |
| ffns.dense.type | dense | codemodelling codeMLP block |
| ffns.dense.d_ff | 12,288 | config.jsonconfig.jsonintermediate_size |
| ffns.dense.gated | true | codemodelling codeMLP: gated (SwiGLU/GeGLU) |
| ffns.moe.type | moe | codemodelling codeMoE block |
| ffns.moe.experts | 160 | config.jsonconfig.jsonn_routed_experts |
| ffns.moe.active | 6 | config.jsonconfig.jsonnum_experts_per_tok |
| ffns.moe.d_expert | 1,536 | config.jsonconfig.jsonmoe_intermediate_size |
| ffns.moe.gated | true | codemodelling codeexperts are gated MLPs |
| ffns.moe.shared | 2 | config.jsonconfig.jsonn_shared_experts |
| ffns.moe.d_shared | 1,536 | config.jsonconfig.jsonmoe_intermediate_size |
| layout | 1× mla/dense · 59× mla/moe | config.jsonconfig.jsonnum_hidden_layers, first_k_dense_replace |
Sources
- config.json @ 4461458
- model card
- arXiv 2405.04434
- modelling code · transformers 5.18.0 modelling code, or the model repository's own modelling file at the pinned revision