趋近智
参数
7B
上下文长度
8K
模态
Text
架构
Dense
许可证
Gemma Terms of Use
发布日期
21 Feb 2024
训练数据截止日期
-
不同量化方法和上下文大小的显存要求
1024 个令牌
消费级
1x RTX 4090
24GB VRAM
数据中心
1x NVIDIA A100
80GB VRAM
Apple Silicon
1x Apple M3 Max
128GB VRAM
8192 个令牌
消费级
1x RTX 4090
24GB VRAM
数据中心
1x NVIDIA A100
80GB VRAM
Apple Silicon
1x Apple M3 Max
128GB VRAM
没有可用的 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 预测头数
-
分词器
词汇量大小
-
Gemma 1 是由 Google 推出的一系列轻量级、仅解码器架构的 Transformer 模型,提供 2B 和 7B 两种参数规模。该系列模型旨在处理各种文本生成任务,集成了旋转位置嵌入、共享输入/输出嵌入、GEGLU 激活函数以及 RMSNorm。其中,2B 模型采用多查询注意力(Multi-Query Attention)机制,而 7B 模型则采用多头注意力(Multi-Head Attention)机制。
APX AI
在线