Qwen · 2024-06 · 稠密

Qwen2 72B

Qwen2 72B 是 Qwen 於 2024-06 發布的 分組查詢注意力(GQA) Transformer,共 80 層,隱藏維度 8192,上下文視窗 32,768 個 token。

層堆疊

注意力 ×80 · 層數 80

關鍵參數

實驗室Qwen
發布時間2024-06
總參數71.4 B
上下文32,768 tokens
注意力GQA (64:8)
層數80
隱藏維度8,192
注意力頭64
詞表152,064
位置編碼RoPE
正規化RMSNorm
激活函數silu
訓練精度bf16
架構類別Qwen2ForCausalLM

架構概覽

t₁ t₂ t₃ t₄ 輸入 token Embedding · 詞表 152,064 → 隱藏維度 8,192 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn 注意力頭 64 KV 頭 8 head dim 128 · RoPE FFN · SwiGLU FFN 維度 29,568 × 80 Transformer 块 注意力 前饋 / MoE RMSNorm pre-norm 最終正規化 · RMSNorm LM head → 詞表 152,064 p p p → 下一 token 位置編碼 RoPE 訓練精度 BF16 上下文 32K tok
依據隨權重發布的 config.json 繪製 · 80 層 / 寬 8,192 / 上下文 32,768。本圖為本站原創示意圖。

注意力

分組查詢注意力(GQA) — 64 q-heads / 8 kv-heads. RoPE θ=1,000,000.

前饋 / MoE

它是稠密模型:全部 71.4B 參數都參與每個 token 的計算。

欄位級對比

對比對象為同實驗室的上一代模型;無前代時取結構最相似的模型。倍數欄 = 本模型數值 ÷ 對比模型數值。

與前代 Qwen1.5 7B 的全欄位對比
模型Qwen2 72BQwen1.5 7B倍數
model_typeqwen2qwen2
architecturesQwen2ForCausalLMQwen2ForCausalLM
hidden_size81924096×2.00
num_hidden_layers8032×2.50
num_attention_heads6432×2.00
num_key_value_heads832×0.250
intermediate_size2956811008×2.69
hidden_actsilusilu
rope_theta10000001000000≈1
max_position_embeddings3276832768≈1
sliding_window13107232768×4.00
rms_norm_eps0.0000010.000001≈1
vocab_size152064151936≈1
torch_dtypebfloat16bfloat16
attention_dropout00
bos_token_id151643151643≈1
eos_token_id151645151643≈1
initializer_range0.020.02≈1
max_window_layers8028×2.86
tie_word_embeddingsfalsefalse
transformers_version4.40.14.37.0
use_cachetruetrue
use_sliding_windowfalsefalse

結構最相似的模型

相似度取值範圍 0(無共同類別特徵)到 1(結構輪廓完全一致)。

原始 config 欄位

23 個欄位
architecturesQwen2ForCausalLM
attention_dropout0
bos_token_id151643
eos_token_id151645
hidden_actsilu
hidden_size8192
initializer_range0.02
intermediate_size29568
max_position_embeddings32768
max_window_layers80
model_typeqwen2
num_attention_heads64
num_hidden_layers80
num_key_value_heads8
rms_norm_eps0.000001
rope_theta1000000
sliding_window131072
tie_word_embeddingsfalse
torch_dtypebfloat16
transformers_version4.40.1
use_cachetrue
use_sliding_windowfalse
vocab_size152064