【随手记录】Llama Tutorial 大语言模型实践

随笔2个月前发布 西川不吃姜
5 0 0

这个tutorial的契机是yy突然看到了一个workshop 所以类似于一周大作业的形式,输入command输出使用了自动驾驶哪些模块,代码在这里 所以就干一干,顺便写一个tutorial给大家参考和教程 引申更多的应用

参考资料:

https://github.com/facebookresearch/codellama, https://github.com/facebookresearch/llama
模型申请地址:https://ai.meta.com/llama/ → 然后按download,填写完后 就会收到两封邮件
https://github.com/oobabooga/text-generation-webui
写tutorial期间的代码:https://github.com/KTH-RPL/DriveCmd_LLM

所有模型参数和所需要的GPU并行,注意这只是模型load占用的memory 如果token或者batch size较大 也需要很多GPU显存,表格下方会说明每个之间的区别,MP是指并行GPU的数量 默认是多少,所有模型我都在 这边服务器的A100试过 Size已经是乘过数量的了 是总和需要的memory

Model Size MP
code-7B ~12.5G 1
code-13B 24G 2
code-34B 63G 4
7b-chat ~12.5G 1
13b-chat 24G 2
70b-chat ~160G 8

7B, 13B, 34B 指示模型参数量
chat 指示经过了对话的fine-tuned 【 llama2论文原文:a fine-tuned version of Llama 2 that is optimized for dialogue use cases】
code 指示此llama版本经过了code数据集的训练,can use text prompts to generate and discuss code. 此处为 官方blog

温馨提示 下列内容和 此 https://github.com/KTH-RPL/DriveCmd_LLM README 有重复:里面包含了example和我们写的prompt流程等 and yy尝试的chatgpt api那边的script

1. 申请下载模型

具体如图:

【随手记录】Llama Tutorial 大语言模型实践

https://ai.meta.com/llama/ 点击download
填写资料,勾上所有的东西
接受terms 然后点continue
大概10分钟内能收到两封邮件,告诉你可用的权重和其下载地址【注意!这个地址有限制比如下载超过3次/24小时后 请重新返回步骤一】

2. 模型部署

正如前言中提到的,大部分本地自己用的电脑 可能只能跑最小的模型也就是7b的 GPU的占用率大概是12G,如下图为我运行repo里的代码时占用截图 【上一个A100试了一下 这么一看我本机3090 也带不动】

内存的增大原因:

prompt 长短 也就是你给模型的输入文字 可能很多 【这也是为什么chatgpt按token收费
输出的语句长短 也就是模型给你的回答 【我这因为让他解释他的输出了 所以也比较长

【随手记录】Llama Tutorial 大语言模型实践

回归正题,接下来是clone两边的代码,一个是code一个是llama 前者经过了八股文、leetcode训练【我是这么理解的 hhh 通俗易懂】 后者是原生态

https://github.com/facebookresearch/llama, 注意复制链接的时候 也别给错了 看对邮件给
https://github.com/facebookresearch/codellama 同上理

以下为演示截图

【随手记录】Llama Tutorial 大语言模型实践

git clone 对应代码
进入文件夹
运行download.sh
copy 邮件里收到的下载链接【第一次没看仔细 一直在填自己的email 可还行
选择下载什么模型,注意全部下载的话 可能比较大 建议硬盘空间不够的 选一个最小的 先试试

3. 模型运行

这一步官方的README写的挺不错的,依赖很少 照着pip install -r requirements.txt就行,基本无障碍,最好是建一个conda的环境,别和系统混了,此处为运行示例:


jsx

torchrun --nproc_per_node 1 example_completion.py 
    --ckpt_dir CodeLlama-7b/ 
    --tokenizer_path CodeLlama-7b/tokenizer.model 
    --max_seq_len 128 --max_batch_size 4

截图运行及GPU占用:

【随手记录】Llama Tutorial 大语言模型实践

正常我们想要的应该是chat 聊天形式的所以选择以下模型会有更好的效果:

Model Size MP
CodeLlama-7b-Instruct ~12.5G 1
CodeLlama-13b-Instruct 24G 2
CodeLlama-34b-Instruct 63G 4
llama-2-7b-chat ~12.5G 1
llama-2-13b-chat 24G 2
llama-2-70b-chat ~160G 8

4. More

可以参考 前言中我提到的task 给出的一个结果调用,这里是任务的报告pdf:https://arxiv.org/abs/2311.08206 更方便大家直接选择和对比,对应代码在 https://github.com/KTH-RPL/DriveCmd_LLM

【随手记录】Llama Tutorial 大语言模型实践

还有就是如果想自己部署一个chatgpt,或者是在苹果系统下(which 内存就是显卡内存 可以参考这个b站视频:苹果M2 Ultra:AI大模型的新希望

自己部署一个chatgpt参考code:

llama or codellama 作为model和basic
text generation web UI 作为一个web的界面 这样才能开启正常对话, https://github.com/oobabooga/text-generation-webui

部署后的示意图:

【随手记录】Llama Tutorial 大语言模型实践

同时这里面也提到了其他的大语言模型 可以尝试的:Multiple model backends: transformers, llama.cpp, ExLlama, ExLlamaV2, AutoGPTQ, GPTQ-for-LLaMa, CTransformers, AutoAWQ

还有一些vision assistance的模型可以一起,免费版chatgpt 4.0 (但是试了一下即使是70B的 效果在我们的任务上也没有 3.5API调用的高,果然还是钱的问题

其中有一个大视觉语言模型 可以关注:https://github.com/haotian-liu/LLaVA

5. Llama2 论文速读

主要是记录一下一些细节 比如模型和模型之间的不同,fine-tuned的实现等。

首先是关于llama2-chat的训练,如下图:得到llama2后 经过RLHF 步骤得到的chat模型 【RLHF 是一种模型训练过程,应用于微调的语言模型,以进一步使模型行为与人类偏好和指令遵循保持一致。】

【随手记录】Llama Tutorial 大语言模型实践

下图说明了 训练的一些细节:

token是指将所有number 拆分成独立的digits 然后使用use bytes 去decompose 未知的UTF-8,总的词汇是32k token
预训练 2T 的token后 模型也没有饱和

【随手记录】Llama Tutorial 大语言模型实践

__EOF__

【随手记录】Llama Tutorial 大语言模型实践
本文作者: Kin Zhang 本文链接: https://www.cnblogs.com/kin-zhang/p/17834075.html 关于博主: 欢迎评论区讨论,私信不一定回复 版权声明: 本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处! 声援博主: 如果您觉得文章对您有帮助,可以点击文章右下角【推荐】一下。

© 版权声明

相关文章

暂无评论

您必须登录才能参与评论!
立即登录
暂无评论...