← Tech Blog

Archive · 2024

基于llam2学习推理过程

在上一节学习了Transformer的搭建过程,但是对于实际推理的流程,比如参数如何加载?batch中的多个sequence如何并行地推理还不甚清楚。由于vLLM内容繁杂,故使用llama提供的generator来做推理过程的学习,顺便再重温一下llama2的…

基于Pytorch复现Transformer

学习LLM之前,首先就是要搞懂Transformer的原理并复现。主要参考了[该文章][2]

cmu 10-414 HW4

本节只展示我认为homework4中比较重要的部分,其余参考GitHub代码