趋近智
参数
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
确定特定工作负载的硬件需求:
在显存计算器中打开
排名
#238
| 基准 | 分数 | 排名 |
|---|---|---|
通用文本 | 1138 | 192 |
排名
#238
编程排名
-
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 是由 Google 推出的一系列轻量级、仅解码器架构的 Transformer 模型,提供 2B 和 7B 两种参数规模。该系列模型旨在处理各种文本生成任务,集成了旋转位置嵌入、共享输入/输出嵌入、GEGLU 激活函数以及 RMSNorm。其中,2B 模型采用多查询注意力(Multi-Query Attention)机制,而 7B 模型则采用多头注意力(Multi-Head Attention)机制。