ApX 标志ApX 标志

趋近智

Gemma 1 7B

参数

7B

上下文长度

8K

模态

Text

架构

Dense

KV 缓存速率

336 KB/tok

许可证

Gemma Terms of Use

发布日期

21 Feb 2024

训练数据截止日期

-

API 价格 (每 1M)

输入: $0.05 · 输出: $0.08

硬件选型

精度

上下文

工作负载显存: 8.38 GB

INT4 · 8K context

基础权重: 3.26 GB

KV 缓存: 2.63 GB

运行时与计算图: 2.50 GB

最小所需硬件

1x NVIDIA L4 24GB

8.4 / 21.6 GB usable

1x RTX 3060 12GB

8.4 / 10.8 GB usable

Apple M4 24GB

8.4 / 18.0 GB usable

确定特定工作负载的硬件需求:

在显存计算器中打开

架构图

Input TokensToken EmbeddingPosition: RoPEHidden: 3.1k · Context: 8Kx 28 layersRMSNormPre-AttentionMulti-Head Attention32Q / 32KV headsHead dim: 96+RMSNormPre-FFNFeed-Forward NetworkActivation+Final RMSNormOutput Logits

评估基准

排名

#238

基准分数排名

通用文本

Text Arena

1138

192

排名

排名

#238

编程排名

-

关于 Gemma 1 7B

Gemma 是由 Google 开发的一个轻量级、仅解码器 (decoder-only) 语言模型系列,借鉴了与 Gemini 模型相同的研究和技术。Gemma 1 7B 是其 70 亿参数变体,专为文本到文本生成任务而设计,包括问答、摘要和推理。该模型采用了 Transformer 仅解码器架构。

关键架构组件包括用于注意力机制的多头注意力 (MHA) 和用于编码位置信息的旋转位置嵌入 (RoPE)。所使用的激活函数为 GeGLU,归一化则通过 RMSNorm 完成。该模型的训练依托于 Google 的第五代张量处理单元 (TPUv5e),并利用 JAX 和 ML Pathways 实现了高效的大规模训练。

Gemma 1 7B 在约 6 万亿个 token 的数据上进行了训练,这些数据主要为英文,涵盖了多样化的网络文档、数学文本和代码。数据预处理过程涉及严格的过滤,以移除有害或敏感内容,符合负责任的 AI 开发实践。该模型相对紧凑的尺寸使其能够部署在从个人笔记本电脑、工作站到云基础设施的各种环境中。

技术规格

注意力

注意力结构

Multi-Head Attention

注意力头

32

键值头

32

位置嵌入

ROPE

归一化

RMS Normalization

维度

隐藏维度大小

3,072

层数

28

关于 Gemma 1

Gemma 1 是由 Google 推出的一系列轻量级、仅解码器架构的 Transformer 模型,提供 2B 和 7B 两种参数规模。该系列模型旨在处理各种文本生成任务,集成了旋转位置嵌入、共享输入/输出嵌入、GEGLU 激活函数以及 RMSNorm。其中,2B 模型采用多查询注意力(Multi-Query Attention)机制,而 7B 模型则采用多头注意力(Multi-Head Attention)机制。


其他 Gemma 1 模型