0


Gemma模型:谷歌的小巨人,让高性能AI触手可及


认识Gemma:谷歌的开放式大型语言模型

谷歌最近发布了一系列开放式大型语言模型,名为Gemma。这是谷歌加强对开源人工智能支持的一部分,也标志着Gemma与Hugging Face生态系统的完美集成。让我们深入了解Gemma及其如何帮助开发者和研究者在各种硬件上高效部署AI模型。

Gemma模型概览

Gemma模型基于谷歌的Gemini技术,提供了两种规模的版本:2B(20亿)参数和7B(70亿)参数。每种规模都包括基础版本和经过指令优化的版本,旨在满足不同硬件平台的需求:

  • gemma-7b:7B参数的基础模型,适用于消费级GPU和TPU。
  • gemma-7b-it:7B参数的指令优化版本,针对GPU和TPU的高效部署。
  • gemma-2b:2B参数的基础模型,可在CPU和移动设备上运行。
  • gemma-2b-it:2B参数的指令优化版本,为CPU和移动设备优化。

所有版本的Gemma模型都能处理高达8K tokens的输入,这意味着它们能够理解和生成长篇幅的文本。

如何获取和使用Gemma模型

Gemma模型已经集成到Hugging Face Hub,您可以轻松访问和使用这些模型。以下是两个模型的链接,供您参考:

  • gemma-2b-it
  • gemma-7b-it

此外,如果您希望在本地使用这些模型,可以通过以下百度网盘链接下载模型文件:

  • 百度网盘链接:Gemma model
  • 提取码:w8m8

环境配置

为了运行Gemma模型,您需要安装特定版本的库。请使用以下命令进行安装:

pip install-U"transformers==4.38.0"--upgrade
pip installtorch==1.13.0 jinja2==3.1.3

与Gemma对话

下面的Python代码演示了如何使用Gemma模型进行文本生成:

# 导入transformers库的AutoTokenizer和pipeline工具,以及PyTorchfrom transformers import AutoTokenizer, pipeline
import torch

# 模型的路径或者模型的标识符
model ="./gemma-2b-it"# 从预训练模型加载分词器
tokenizer = AutoTokenizer.from_pretrained(model)# 创建一个文本生成pipeline,这里使用的是text-generation任务,指定模型、数据类型及运行设备
pipeline = pipeline("text-generation",# 指定pipeline的任务为文本生成
    model=model,# 使用的模型
    model_kwargs={"torch_dtype": torch.bfloat16},# 指定模型使用的数据类型,这里使用bfloat16以节省内存# model_kwargs={"torch_dtype": torch.bfloat16, "quantization_config": {"load_in_4bit": True}}, # 以 8 位或 4 位模式加载。以 4 位模式加载模型大约需要 9 GB 的内存# 指定模型使用的数据类型,这里使用bfloat16以节省内存
    device="cpu",# 指定运行设备为CPU,如果有GPU可改为"cuda")# 开始对话的提示信息print("----------开始与Gemma对话----------")
history_dialogue =""# 初始化历史对话变量,用于累积对话历史whileTrue:# 开始一个无限循环,直到用户决定退出
    input_text =input("<Human>: ")# 获取用户输入if input_text =="new dialogue":# 如果用户输入"new dialogue",则重置对话历史print("----------重新开始与Gemma对话----------")
        history_dialogue =""# 重置历史对话变量continueelif input_text =="q":# 如果用户输入"q",则退出循环break
    messages =[{"role":"user","content": input_text},# 构造当前轮对话的消息]# 使用pipeline的tokenizer处理消息,生成模型的输入prompt
    prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
    history_dialogue = history_dialogue + prompt  # 更新历史对话变量,累积对话历史# 确保history_dialogue的token长度不超过8k,这是因为模型的限制iflen(tokenizer(history_dialogue)["input_ids"])>8192:print("历史对话过长,无法处理。")break# 使用pipeline生成回复
    outputs = pipeline(
        history_dialogue,
        max_new_tokens=4096,# 最大新生成token数
        add_special_tokens=True,# 添加特殊token
        do_sample=True,# 启用采样
        temperature=0.7,# 采样温度
        top_k=50,# Top-K采样
        top_p=0.95# Top-P采样(nucleus sampling))# 提取模型生成的文本作为回复,去除已有的历史对话部分
    reply = outputs[0]["generated_text"][len(history_dialogue):]print(f"<Gemma>: {reply}")# 打印模型的回复
    prompt = reply +"<end_of_turn>"# 准备下一轮对话的promptprint("对话结束.")# 用户退出对话循环后的提示

请注意,为了保证性能和避免内存溢出,

history_dialogue

的token长度不应超过8192。

对话示例:

gemma对话示例

Gemma模型的主要成就

Gemma模型的推出标志着谷歌在大型语言模型(LLM)领域的一个重要进步,特别是在模型效率和灵活性方面。这些成就主要表现在三个方面:

模型尺寸的优化

Gemma提供了两种不同规模的模型:2B(20亿参数)和7B(70亿参数),这些模型的尺寸相对较小,特别是当考虑到它们强大的性能时。这种尺寸上的优化使得Gemma模型能够在不牺牲性能的情况下运行在硬件资源更受限的环境中。

强大的性能

尽管Gemma模型的参数数量较少,但它们依然能够提供与更大模型相匹配的性能。这得益于谷歌对模型结构和训练过程的优化,包括使用了指令优化版本(

-it

),进一步提升了模型处理特定任务的能力。

本地电脑部署

Gemma模型的一个显著特点是它们可以直接部署在本地电脑上,不只限于云端或高性能服务器。这使得AI研究和开发更加便捷,降低了入门门槛。对于开发者和研究人员而言,能够在本地机器上快速测试和迭代模型是一个巨大的优势,特别是在资源有限或需要快速原型开发的场景中。

结论

Gemma模型凭借其优化的尺寸、强大的性能以及出色的部署灵活性,在AI领域树立了新的标杆。这些模型不仅支持了开源AI社区的发展,还为在各种硬件上部署高效、强大的AI应用提供了更多可能性。无论您是AI领域的专业人士还是热情爱好者,Gemma模型都值得您深入探索和利用。



本文转载自: https://blog.csdn.net/qq_39045712/article/details/136241133
版权归原作者 一花·一树 所有, 如有侵权,请联系我们删除。

“Gemma模型:谷歌的小巨人,让高性能AI触手可及”的评论:

还没有评论