← 返回主题列表
Q
QianXun
@QianXun · 2026年06月13日 13:39 · 6浏览

《Born》第18章:文本生成——Tokenizer、采样策略与流式输出

从 logits 到文字,这是一条完整的生成管线。这一章拆解 Born 的文本生成系统。

---

生成管线全景

Prompt → Tokenizer.Encode → tokens
  → for i in range(maxTokens):
      logits = model.Forward(tokens, kvCache)
      nextToken = Sampler.Sample(logits[-1])
      tokens = append(tokens, nextToken)
      if stopConditionMet: break
  → Tokenizer.Decode(tokens) → 生成的文本

---

Tokenizer

// TikToken(GPT 系列)
tokenizer := tiktoken.New("cl100k_base")
tokens := tokenizer.Encode("Hello, world!")
// [15496, 11, 995, 0]

text := tokenizer.Decode(tokens)
// "Hello, world!"

Born 支持两种 tokenizer:TikToken(GPT/Claude 系列)和 BPE(BERT 系列)。

---

采样策略

sampler := nn.NewSampler(nn.SamplerConfig{
    Temperature: 0.8,    // 降低随机性
    TopK: 40,            // 只保留概率最高的 40 个 token
    TopP: 0.95,          // 累积概率达到 95% 时截断
    MinP: 0.05,          // 最低相对概率阈值
})

nextToken := sampler.Sample(logits)

参数作用典型值
Temperature控制随机性0.7-1.0
TopK限制候选集大小20-50
TopP动态截断0.9-0.95
MinP过滤低概率噪声0.01-0.05
---

流式输出

// 逐 token 输出,不等待整段生成完毕
for token := range generator.Generate(prompt) {
    text := tokenizer.Decode([]int{token})
    fmt.Print(text) // 实时输出
    os.Stdout.Sync()
}

---

📘 《Born》连载技术书,第 18/22 章。

👍 1
💬 讨论回复 (0)
推荐

🌟 智谱 GLM-5 已上线

我正在智谱大模型开放平台 BigModel.cn 上打造 AI 应用,智谱新一代旗舰模型 GLM-5 已上线,在推理、代码、智能体综合能力达到开源模型 SOTA 水平。

🎁 领取 2000万 Tokens