ApX 标志ApX 标志

趋近智

Gemma 1 7B

参数

7B

上下文长度

8K

模态

Text

架构

Dense

许可证

Gemma Terms of Use

发布日期

21 Feb 2024

训练数据截止日期

-

系统要求

不同量化方法和上下文大小的显存要求

1024 个令牌

16.69 GB VRAM

消费级

1x RTX 4090

24GB VRAM

数据中心

1x NVIDIA A100

80GB VRAM

Apple Silicon

1x Apple M3 Max

128GB VRAM

8192 个令牌

20.15 GB VRAM

消费级

1x RTX 4090

24GB VRAM

数据中心

1x NVIDIA A100

80GB VRAM

Apple Silicon

1x Apple M3 Max

128GB VRAM

架构图

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

评估基准

没有可用的 Gemma 1 7B 评估基准。

排名

排名

-

编程排名

-

关于 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

RoPE Theta

-

滑动窗口注意力

-

滑动窗口大小

-

滑动窗口比例

-

线性注意力

-

线性注意力比例

-

归一化

RMS Normalization

激活函数

-

维度

隐藏维度大小

3,072

层数

28

FFN 中间层大小(稠密层)

-

多 Token 预测头数

-

分词器

词汇量大小

-

模型完整性

总分

B-

62 / 100

关于 Gemma 1

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


其他 Gemma 1 模型
Gemma 1 7B:规格和 GPU 显存要求