阅读llama源码笔记_1

文章目录

总体逻辑

以如下命令为例,梳理一下总体流程

bash 复制代码
torchrun --nproc_per_node 1 example_text_completion.py \
    --ckpt_dir llama-2-7b-chat/ \
    --tokenizer_path tokenizer.model \
    --max_seq_len 512 --max_batch_size 6

在顶层目录运行使用torchrun运行example_text_completion.py文件,example_text_completion.py中逻辑可分为三大步:

1) 构造模型

generator = Llama.build(
        ckpt_dir=ckpt_dir,
        tokenizer_path=tokenizer_path,
        max_seq_len=max_seq_len,
        max_batch_size=max_batch_size,
    )

2) 定义数据

prompts: List[str] = [
        # For these prompts, the expected answer is the natural continuation of the prompt
        "I believe the meaning of life is",
        "Simply put, the theory of relativity states that ",
        """A brief message congratulating the team on the launch:

        Hi everyone,
        
        I just """,
        # Few shot prompt (providing a few examples before asking model to complete more);
        """Translate English to French:
        
        sea otter => loutre de mer
        peppermint => menthe poivrée
        plush girafe => girafe peluche
        cheese =>""",
    ]

3) 处理下游任务

对不同的数据有不同的处理方法,对相同的数据也有不同的任务,text_completion函数处理的下游任务是文本生成

c 复制代码
results = generator.text_completion(
        prompts,
        max_gen_len=max_gen_len,
        temperature=temperature,
        top_p=top_p,
    )

接下来对三个步骤进行分析,只列出关键代码

1.构造模型

Llama.build函数是llama/llama/目录下generation.py文件中定义的,build最开始的一些并行参数设置就不分析了,因为我目前也不是很懂,在单GPU上跑可以先不考虑并行的相关操作。

build函数构建了一个tokenizer模型(分词器)和一个Transformer模型,分别对应tokenizer.py 文件和model.py文件,其实也很好理解,输入的字符串要先过分词器变为tokenid再输入给Transformer

build函数逻辑上也可以分为2步:

1.1构造分词器tokenizer

c 复制代码
tokenizer = Tokenizer(model_path=tokenizer_path)

tokenizer是放在CPU上的

tokenizer.encode()将一个字符串编码成一串tokenId

tokenizer.decode() 一串tokenId解码成字符串

1.2构造Transformer

先调用torch.load将权重参数加载到CPU上,然后在GPU上初始化Transformer模型,最后再调用load_state_dict将CPU上的参数传输到GPU赋值给Transformer模型

c 复制代码
checkpoint = torch.load(ckpt_path, map_location="cpu")//将权重参数加载到CPU上
torch.set_default_tensor_type(torch.cuda.HalfTensor)
model = Transformer(model_args)//在GPU上初始化Transformer模型    底层调用cudaMalloc在GPU开辟参数存放空间,并使用FillFunctor将参数都初始化为0
model.load_state_dict(checkpoint, strict=False)//将CPU上的参数复制到GPU赋值给Transformer模型  底层成对调用cudaMemcpyAsync和cudaStreamSynchronize

底层的调用关系我是通过nsys computer进行分析的

2.定义数据

没啥好讲的,声明一个 List[str]对象,当然也可以从网上下载其他数据集,如果真正线上部署的话,那么就是通过网络请求传来的数据了

3.处理下游任务

将数据通过generator.text_completion函数(generation.py中定义的)处理,text_completion函数逻辑如下:

3.1分词(编码)

将字符串编码为tokenid

c 复制代码
prompt_tokens = [self.tokenizer.encode(x, bos=True, eos=False) for x in prompts]

3.2 generate

该函数逻辑后面再分析,函数的返回值就是生成的tokenId

c 复制代码
generation_tokens, generation_logprobs = self.generate(
            prompt_tokens=prompt_tokens,
            max_gen_len=max_gen_len,
            temperature=temperature,
            top_p=top_p,
            logprobs=logprobs,
            echo=echo,
        )

3.3 解码

即将tokenId解码成字符串,即该下游任务的最终结果

c 复制代码
self.tokenizer.decode(t)} for t in generation_tokens
相关推荐
密码小丑1 小时前
11月4日(内网横向移动(一))
笔记
鸭鸭梨吖1 小时前
产品经理笔记
笔记·产品经理
齐 飞2 小时前
MongoDB笔记01-概念与安装
前端·数据库·笔记·后端·mongodb
丫头,冲鸭!!!2 小时前
B树(B-Tree)和B+树(B+ Tree)
笔记·算法
听忆.3 小时前
手机屏幕上进行OCR识别方案
笔记
SpikeKing3 小时前
LLM - 使用 LLaMA-Factory 微调大模型 环境配置与训练推理 教程 (1)
人工智能·llm·大语言模型·llama·环境配置·llamafactory·训练框架
Selina K3 小时前
shell脚本知识点记录
笔记·shell
4 小时前
开源竞争-数据驱动成长-11/05-大专生的思考
人工智能·笔记·学习·算法·机器学习
霍格沃兹测试开发学社测试人社区4 小时前
软件测试学习笔记丨Flask操作数据库-数据库和表的管理
软件测试·笔记·测试开发·学习·flask
幸运超级加倍~5 小时前
软件设计师-上午题-16 算法(4-5分)
笔记·算法