Gemma 3 1b Pt

🚀 Gemma 3 模型卡片

Gemma 3 是谷歌推出的轻量级多模态模型，基于与Gemini相同的技术构建。它支持文本和图像输入，输出文本，具有 128K 大上下文窗口和超 140 种语言支持，适用于多种文本生成和图像理解任务。

🚀 快速开始

安装

要运行 Gemma 3 模型，首先需要安装 transformers 库。Gemma 3 从 transformers 4.50.0 版本开始支持。

$ pip install -U transformers

运行示例

使用 pipeline API 运行

from transformers import pipeline
import torch

pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16)
output = pipe("Eiffel tower is located in", max_new_tokens=50)

在单/多 GPU 上运行模型

import torch
from transformers import AutoTokenizer, Gemma3ForCausalLM

ckpt = "google/gemma-3-1b-pt"
tokenizer = AutoTokenizer.from_pretrained(ckpt)
model = Gemma3ForCausalLM.from_pretrained(
    ckpt,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)

prompt = "Eiffel tower is located in"
model_inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

input_len = model_inputs["input_ids"].shape[-1]

with torch.inference_mode():
    generation = model.generate(**model_inputs, max_new_tokens=50, do_sample=False)
    generation = generation[0][input_len:]

decoded = tokenizer.decode(generation, skip_special_tokens=True)
print(decoded)

✨ 主要特性

• 多模态处理：支持文本和图像输入，输出文本，适用于多种文本生成和图像理解任务。
• 大上下文窗口：拥有 128K 的大上下文窗口，能够处理更长的输入。
• 多语言支持：支持超过 140 种语言，具有更广泛的应用场景。
• 轻量化设计：相对较小的模型尺寸，可在资源有限的环境中部署，如笔记本电脑、台式机或自有云基础设施。

📦 安装指南

要使用 Gemma 3 模型，需要安装 transformers 库。Gemma 3 从 transformers 4.50.0 版本开始支持。

$ pip install -U transformers

💻 使用示例

基础用法

from transformers import pipeline
import torch

pipe = pipeline("text-generation", model="google/gemma-3-1b-pt", device="cuda", torch_dtype=torch.bfloat16)
output = pipe("Eiffel tower is located in", max_new_tokens=50)

高级用法

import torch
from transformers import AutoTokenizer, Gemma3ForCausalLM

ckpt = "google/gemma-3-1b-pt"
tokenizer = AutoTokenizer.from_pretrained(ckpt)
model = Gemma3ForCausalLM.from_pretrained(
    ckpt,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)

prompt = "Eiffel tower is located in"
model_inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

input_len = model_inputs["input_ids"].shape[-1]

with torch.inference_mode():
    generation = model.generate(**model_inputs, max_new_tokens=50, do_sample=False)
    generation = generation[0][input_len:]

decoded = tokenizer.decode(generation, skip_special_tokens=True)
print(decoded)

📚 详细文档

模型信息

描述

Gemma 是谷歌推出的一系列轻量级、先进的开放模型，基于与 Gemini 模型相同的研究和技术构建。Gemma 3 模型是多模态的，能够处理文本和图像输入，并生成文本输出。预训练和指令调优变体的权重均开放。Gemma 3 具有 128K 的大上下文窗口，支持超过 140 种语言，并且比以前的版本有更多的尺寸可供选择。Gemma 3 模型非常适合各种文本生成和图像理解任务，包括问答、摘要和推理。其相对较小的尺寸使其能够在资源有限的环境中部署，如笔记本电脑、台式机或自有云基础设施，使先进的人工智能模型更易于获取，促进创新。

输入和输出

引用

@article{gemma_2025,
    title={Gemma 3},
    url={https://goo.gle/Gemma3Report},
    publisher={Kaggle},
    author={Gemma Team},
    year={2025}
}

模型数据

训练数据集

这些模型在包含多种来源的文本数据集上进行训练。27B 模型使用 14 万亿个标记进行训练，12B 模型使用 12 万亿个标记，4B 模型使用 4 万亿个标记，1B 模型使用 2 万亿个标记。主要组成部分如下：

• 网络文档：多样化的网络文本集合确保模型接触到广泛的语言风格、主题和词汇。训练数据集包含超过 140 种语言的内容。
• 代码：让模型接触代码有助于学习编程语言的语法和模式，提高生成代码和理解代码相关问题的能力。
• 数学：在数学文本上进行训练有助于模型学习逻辑推理、符号表示和处理数学查询。
• 图像：广泛的图像使模型能够执行图像分析和视觉数据提取任务。

数据预处理

对训练数据应用了以下关键的数据清理和过滤方法：

• CSAM 过滤：在数据准备过程的多个阶段应用了严格的 CSAM（儿童性虐待材料）过滤，以确保排除有害和非法内容。
• 敏感数据过滤：作为使 Gemma 预训练模型安全可靠的一部分，使用自动化技术从训练集中过滤出某些个人信息和其他敏感数据。
• 其他方法：根据内容质量和安全性进行过滤，符合我们的政策。

实现信息

硬件

Gemma 使用 张量处理单元 (TPU) 硬件（TPUv4p、TPUv5p 和 TPUv5e）进行训练。训练视觉语言模型 (VLM) 需要大量的计算能力。TPU 专门为机器学习中常见的矩阵运算而设计，在这一领域具有以下优势：

• 性能：TPU 专门用于处理训练 VLM 涉及的大量计算。与 CPU 相比，它们可以显著加速训练。
• 内存：TPU 通常配备大量的高带宽内存，允许在训练期间处理大型模型和批量大小。这有助于提高模型质量。
• 可扩展性：TPU Pod（大型 TPU 集群）为处理大型基础模型日益增长的复杂性提供了可扩展的解决方案。可以在多个 TPU 设备上分布训练，以实现更快、更高效的处理。
• 成本效益：在许多情况下，与基于 CPU 的基础设施相比，TPU 可以为训练大型模型提供更具成本效益的解决方案，特别是考虑到更快的训练节省的时间和资源。
• 这些优势符合 谷歌对可持续运营的承诺。

软件

使用 JAX 和 ML Pathways 进行训练。

JAX 使研究人员能够利用最新一代的硬件（包括 TPU），更快、更高效地训练大型模型。ML Pathways 是谷歌构建能够跨多个任务进行泛化的人工智能系统的最新努力。这特别适用于基础模型，包括此类大型语言模型。

JAX 和 ML Pathways 结合使用，如 关于 Gemini 模型家族的论文 所述：“Jax 和 Pathways 的‘单控制器’编程模型允许单个 Python 进程编排整个训练运行，大大简化了开发工作流程。”

评估

基准测试结果

这些模型针对大量不同的数据集和指标进行了评估，以涵盖文本生成的不同方面。

推理和事实性

STEM 和代码

多语言

多模态

伦理和安全

评估方法

我们的评估方法包括结构化评估和对相关内容政策的内部红队测试。红队测试由多个不同的团队进行，每个团队有不同的目标和人类评估指标。这些模型针对与伦理和安全相关的多个不同类别进行了评估，包括：

• 儿童安全：评估覆盖儿童安全政策的文本到文本和图像到文本提示，包括儿童性虐待和剥削。
• 内容安全：评估覆盖安全政策的文本到文本和图像到文本提示，包括骚扰、暴力和血腥内容以及仇恨言论。
• 代表性危害：评估覆盖安全政策的文本到文本和图像到文本提示，包括偏见、刻板印象和有害关联或不准确信息。

除了开发阶段的评估，我们还进行“保证评估”，这是我们独立的内部评估，用于责任治理决策。它们与模型开发团队分开进行，为发布决策提供信息。高级别结果反馈给模型团队，但提示集被保留，以防止过拟合并确保结果能够为决策提供信息。保证评估结果作为发布审查的一部分报告给我们的责任与安全委员会。

评估结果

在所有安全测试领域，与之前的 Gemma 模型相比，我们在儿童安全、内容安全和代表性危害类别中看到了重大改进。所有测试均在没有安全过滤器的情况下进行，以评估模型的能力和行为。对于文本到文本和图像到文本，以及所有模型尺寸，模型产生的政策违规极少，并且在无根据推断方面比之前的 Gemma 模型表现出显著改进。我们评估的一个局限性是只包括英语提示。

使用和限制

预期用途

开放视觉语言模型 (VLM) 在各个行业和领域有广泛的应用。以下潜在用途列表并非详尽无遗。此列表的目的是提供有关模型创建者在模型训练和开发过程中考虑的可能用例的上下文信息。

• 内容创作和通信
  • 文本生成：这些模型可用于生成创意文本格式，如诗歌、脚本、代码、营销文案和电子邮件草稿。
  • 聊天机器人和对话式 AI：为客户服务、虚拟助手或交互式应用程序提供对话界面。
  • 文本摘要：生成文本语料库、研究论文或报告的简洁摘要。
  • 图像数据提取：这些模型可用于提取、解释和总结视觉数据，用于文本通信。
• 研究和教育
  • 自然语言处理 (NLP) 和 VLM 研究：这些模型可以作为研究人员试验 VLM 和 NLP 技术、开发算法和推动该领域发展的基础。
  • 语言学习工具：支持交互式语言学习体验，有助于语法纠正或提供写作练习。
  • 知识探索：帮助研究人员通过生成摘要或回答特定主题的问题来探索大量文本。

限制

• 训练数据
  • 训练数据的质量和多样性显著影响模型的能力。训练数据中的偏差或差距可能导致模型响应的局限性。
  • 训练数据集的范围决定了模型能够有效处理的主题领域。
• 上下文和任务复杂性
  • 模型更擅长可以用清晰提示和指令构建的任务。开放式或高度复杂的任务可能具有挑战性。
  • 模型的性能可能受到提供的上下文量的影响（在一定程度上，更长的上下文通常会导致更好的输出）。
• 语言歧义与细微差别
  • 自然语言本质上是复杂的。模型可能难以理解微妙的细微差别、讽刺或比喻语言。
• 事实准确性
  • 模型根据从训练数据中学到的信息生成响应，但它们不是知识库。它们可能生成不正确或过时的事实陈述。
• 常识
  • 模型依赖于语言中的统计模式。它们可能在某些情况下缺乏应用常识推理的能力。

伦理考虑和风险

视觉语言模型 (VLM) 的开发引发了一些伦理问题。在创建开放模型时，我们仔细考虑了以下方面：

• 偏差和公平性
  • 在大规模真实世界文本和图像数据上训练的 VLM 可能反映训练材料中嵌入的社会文化偏差。这些模型经过了仔细审查，输入数据预处理在本卡片中描述，后验评估结果也有报告。
• 错误信息和滥用
  • VLM 可能被滥用来生成虚假、误导或有害的文本。
  • 提供了负责任使用模型的指南，请参阅 负责任生成式 AI 工具包。
• 透明度和问责制
  • 本模型卡片总结了模型架构、能力、限制和评估过程的详细信息。
  • 负责任开发的开放模型为开发者和研究人员提供了共享创新的机会，使 VLM 技术在人工智能生态系统中更易获取。

风险识别和缓解措施

• 偏差的延续：鼓励在模型训练、微调及其他用例中进行持续监测（使用评估指标、人工审查）并探索去偏技术。
• 有害内容的生成：内容安全机制和指南至关重要。鼓励开发者根据其特定的产品政策和应用用例谨慎行事并实施适当的内容安全保障措施。
• 恶意使用：技术限制以及对开发者和最终用户的教育有助于减轻 VLM 的恶意应用。提供教育资源和用户举报滥用的机制。Gemma 模型的禁止使用情况在 Gemma 禁止使用政策 中列出。
• 隐私侵犯：模型在经过过滤以去除某些个人信息和其他敏感数据的数据上进行训练。鼓励开发者遵守隐私法规并采用隐私保护技术。

优势

在发布时，与类似规模的模型相比，这个模型家族提供了从头开始为负责任的人工智能开发设计的高性能开放视觉语言模型实现。

使用本文档中描述的基准评估指标，这些模型已显示出比其他类似规模的开放模型替代方案具有更好的性能。

其他信息

• 模型页面：Gemma
• 资源和技术文档：
  • Gemma 3 技术报告
  • 负责任生成式 AI 工具包
  • Kaggle 上的 Gemma
  • Vertex 模型花园中的 Gemma 3
• 使用条款：条款
• 作者：Google DeepMind