(推文施工中)vllm+qlora本地部署微调1.5B大模型
记述了在wsl2中使用vllm部署1.5B的Qwen架构模型,并使用qlora方法进行微调的过程。
(推文施工中)vllm+qlora本地部署微调1.5B大模型
1.目标与流程
1️⃣目标
– 在本地跑起来一个大模型
– 希望使用全量大模型而不是量化过的(ollama就是,所以可以跑7B)
– 熟悉vllm部署流程与操作(实践中常用此框架)
2.环境部署
1️⃣配置要求
内存:32GB
显卡:4060ti 8GB
开启wsl2,使用ubuntu发行版运行。
2️⃣安装环境
- 使用vscode连接ubuntu之后,进入到工作目录。
- 建议创建conda环境:
conda create -n vllm python=3.10 conda activate vllm pip install -r requirements.txt```requirements.txt
1
2
3
3. 从huggingface下载模型:
```cmd
git clone *****
3.运行模型
运行如下命令。注意调整参数以适应电脑配置。(这里不知道为什么怎么调gpu都是占用率拉满…)
python -m vllm.entrypoints.openai.api_server \ #运行vllm服务
--model ~/deepseek1.5B/DeepSeek-R1-Distill-Qwen-1.5B \ #指定模型路径,该路径下需要有模型权重文件
--port 8000 \
--gpu-memory-utilization 0.7 \ #最高gpu利用率,设低了就跑不了
--max-token-len=8192 #最大推理token长度,设高了也跑不了
4.微调与应用
1.使用vllm本地部署DeepSeek-R1-Distill-Qwen-1.5B,这是一个 distill蒸馏模型,基于Qwen架构(一种类transformer架构)
2.在wsl2环境中进行部署。首先从huggingface使用git clone下载模型文件,然后复制进ubuntu, 接下来创建新的conda环境vllm,pip install vllm,版本号见requirements.txt
3. qlora流程:
- 准备数据(指令+输出格式,JSON or JSONL)
- 加载 base 模型(DeepSeek-R1)
- 应用 LoRA 配置(PEFT + bitsandbytes)
- 使用 Trainer 或 SFT Trainer 微调模型
- 保存 adapter 权重(adapter_model)
- 使用 vLLM 加载 base 模型 + adapter
这里的lora和qlora有啥区别?是不是计算量的区别?
4.部署qlora微调需要: –新开一个环境 –准备好json格式的数据(带标签) –
This post is licensed under CC BY 4.0 by the author.