Meta开源大模型LLama2部署使用保姆级教程，附模型对话效果

最新推荐文章于 2024-08-09 07:46:32 发布

AI大模型-王哥

最新推荐文章于 2024-08-09 07:46:32 发布

阅读量508

点赞数 3

文章标签：人工智能 langchain AI大模型大模型

本文链接：https://blog.csdn.net/2401_84204413/article/details/140086162

版权

1.写在前面

就在7月19日，MetaAI开源了LLama2大模型，Meta 首席科学家、图灵奖获得者 Yann LeCun在推特上表示Meta 此举可能将改变大模型行业的竞争格局。一夜之间，大模型格局再次发生巨变。

在这里插入图片描述

推文上列了Llama2的网站和论文，小卷给大家贴一下，感兴趣的友友可以自己看看

2.LLama2是什么

Llama官网的说明是Llama2下一代开源大语言模型，可免费用于学术研究或商业用途。

目前模型有7B、13B、70B三种规格，预训练阶段使用了2万亿Token，SFT阶段使用了超过10w数据，人类偏好数据超过100w。

在这里插入图片描述

另外大家最关心的Llama2和ChatGPT模型的效果对比，在论文里也有提到，

对比GPT-4，Llama2评估结果更优，绿色部分表示Llama2优于GPT4的比例

在这里插入图片描述

虽然中文的占比只有0.13%，但后续会有一大推中文扩充词表预训练&领域数据微调的模型被国人放出。这不才开源几天而已，GIthub上就已经有基于Llama2的中文大模型了。。。

3.部署使用

关于LLama2的技术细节就不再多说了，大家可以自行查阅。接下来就教大家怎么自己玩一玩LLama2对话大模型。

大部分人都是没有本地GPU算力的，我们选择在云服务器上部署使用。我这里用的是揽睿星舟平台的GPU服务器（便宜好用，3090只要1.9/小时，且已在平台上预设了模型文件，无需再次下载）

3.1新建空间

创建一个工作空间，运行环境镜像挂载公有镜像：pytorch: official-torch2.0-cu1117。选择预训练模型：llama-2-7b 和 llama-2-7b-chat。然后创建实例

在这里插入图片描述

3.2下载代码

实例创建完成后，以jupyterLab方式登录服务器，新建一个Terminal，然后进入到data目录下

cd data

下载代码

执行下面的命令从GIthub上拉取llama的代码

sudo git clone https://github.com/facebookresearch/llama.git

下载完成后，会多一个llama目录

在这里插入图片描述

3.3运行脚本

进入llama目录

cd llama

安装依赖

sudo pip install -e .

测试llama-2-7b模型的文本补全能力

命令行执行：

torchrun --nproc_per_node 1 example_text_completion.py \
     --ckpt_dir ../../imported_models/llama-2-7b/Llama-2-7b \
     --tokenizer_path ../../imported_models/llama-2-7b/Llama-2-7b/tokenizer.model \
     --max_seq_len 128 --max_batch_size 4

文本补齐效果示例：

在这里插入图片描述

上面的例子是在python脚本里写了一段话，让模型补全后面的内容。

测试llama-2-7b模型的对话能力

修改llama目录权限为777，再修改example_chat_completion.py文件中的ckpt_dir和tokenizer_path路径为你的llama-2-7b-chat模型的绝对路径

// 1.修改目录权限为可写入
chmod 777 llama

//2.修改example_chat_completion.py文件里的参数
ckpt_dir: str = "/home/user/imported_models/llama-2-7b-chat/Llama-2-7b-chat/",
tokenizer_path: str = "/home/user/imported_models/llama-2-7b-chat/Llama-2-7b-chat/tokenizer.model"

//3.运行对话脚本
torchrun --nproc_per_node 1 example_chat_completion.py

在这里插入图片描述

这里我修改提示语让它用中文回答，执行对话脚本后，对话效果如下：

torchrun --nproc_per_node 1 example_chat_completion.py

在这里插入图片描述

说明：目前官方还没有提供UI界面或是API脚本代码给咱使用，还没法进行对话交互，如果有懂python的友友，可以自行加个UI界面，欢迎大家留言讨论。

4.下载更多模型

llama代码里有download.sh脚本可以下载其他模型，但是下载需要的URL需要自行获取。下载步骤如下：

1.Meta AI网站获取下载URL

MetaAI下载模型页地址：[ai.meta.com/llama/#down…]
在这里插入图片描述

点击Download后，要求填入一些信息和邮箱，提交后会给你的邮箱发一个下载URL，注意这个是你自己的下载链接哦~

下图是小卷邮箱里收到的模型下载链接

在这里插入图片描述

2.下载模型

服务器上命令行执行

sudo bash download.sh

接着按照提示粘贴下载URL和选择要下载的模型

在这里插入图片描述

总结

对于国内大模型使用来说，随着开源可商用的模型越来越多，国内大模型肯定会再次迎来发展机遇。

如何学习AI大模型？

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

在这里插入图片描述