DeepSeek · 2023-11 · 稠密

DeepSeek-Coder 6.7B

DeepSeek-Coder 6.7B 是 DeepSeek 于 2023-11 发布的 多头注意力(MHA) Transformer,共 32 层,隐藏维度 4096,上下文窗口 16,384 个 token。

层堆栈

注意力 ×32 · 层数 32

关键参数

实验室DeepSeek
发布时间2023-11
总参数6.7 B
上下文16,384 tokens
注意力MHA (32:32)
层数32
隐藏维度4,096
注意力头32
词表32,256
位置编码RoPE
归一化RMSNorm
激活函数silu
训练精度bf16
架构类名LlamaForCausalLM

架构概览

t₁ t₂ t₃ t₄ 输入 token Embedding · 词表 32,256 → 隐藏维度 4,096 Q1 Q2 Q3 Qn KV1 KV2 KV3 KVn 注意力头 32 KV 头 32 head dim 128 · RoPE FFN · SwiGLU FFN 维度 11,008 × 32 Transformer 块 注意力 前馈 / MoE RMSNorm pre-norm 最终归一化 · RMSNorm LM head → 词表 32,256 p p p → 下一 token 位置编码 RoPE 训练精度 BF16 上下文 16K tok
依据随权重发布的 config.json 绘制 · 32 层 / 宽 4,096 / 上下文 16,384。本图为本站原创示意图。

注意力

多头注意力(MHA) — 32 q-heads / 32 kv-heads. RoPE θ=100,000.

前馈 / MoE

它是稠密模型:全部 6.7B 参数都参与每个 token 的计算。

字段级对比

对比对象为同实验室的上一代模型;无前代时取结构最相似的模型。倍数列 = 本模型数值 ÷ 对比模型数值。

与最相似 Qwen1.5 7B 的全字段对比
模型DeepSeek-Coder 6.7BQwen1.5 7B倍数
model_typellamaqwen2
architecturesLlamaForCausalLMQwen2ForCausalLM
hidden_size40964096≈1
num_hidden_layers3232≈1
num_attention_heads3232≈1
num_key_value_heads3232≈1
intermediate_size1100811008≈1
hidden_actsilusilu
rope_theta1000001000000×0.100
max_position_embeddings1638432768×0.500
sliding_window32768
rms_norm_eps0.0000010.000001≈1
vocab_size32256151936×0.212
torch_dtypebfloat16bfloat16
bos_token_id32013151643×0.211
eos_token_id32014151643×0.211
initializer_range0.020.02≈1
pretraining_tp1
tie_word_embeddingsfalsefalse
transformers_version4.34.14.37.0
use_cachetruetrue

结构最相似的模型

相似度取值范围 0(无共同类别特征)到 1(结构画像完全一致)。

原始 config 字段

20 个字段
architecturesLlamaForCausalLM
bos_token_id32013
eos_token_id32014
hidden_actsilu
hidden_size4096
initializer_range0.02
intermediate_size11008
max_position_embeddings16384
model_typellama
num_attention_heads32
num_hidden_layers32
num_key_value_heads32
pretraining_tp1
rms_norm_eps0.000001
rope_theta100000
tie_word_embeddingsfalse
torch_dtypebfloat16
transformers_version4.34.1
use_cachetrue
vocab_size32256