返回顶部
热门问答 更多热门问答
技术文章 更多技术文章

WeClone: 用微信聊天记录微调大语言模型

[复制链接]
链载Ai 显示全部楼层 发表于 昨天 20:49 |阅读模式 打印 上一主题 下一主题


通过个人微信聊天记录对大语言模型(LLM)进行微调,打造专属的数字分身。它提供从聊天数据到模型微调、从文本生成到语音克隆、从训练到部署的全链路解决方案。让你的数字分身不仅“会说你的话”,还能“听起来像你本人”。

主要功能

  • 聊天记录训练:用微信聊天记录微调大语言模型,模仿用户说话方式。

  • 高质量语音克隆:用0.5B参数模型和5秒语音样本,生成相似度高达95%的声音。

  • 微信机器人绑定:将数字分身接入微信,支持自动文字和语音回复。

  • 数据预处理工具:提供脚本将聊天记录转为训练数据,默认过滤敏感信息。

  • 模型个性化优化:支持LoRA微调技术,让模型更符合用户语言习惯。

安装和使用

硬件要求

目前项目默认使用chatglm3-6b模型,LoRA方法对sft阶段微调,大约需要16GB显存。也可以使用LLaMA Factory支持的其他模型和方法,占用显存更少,需要自行修改模板的systemprompt词等相关配置。

需要显存的估算值:

方法
精度
7B
14B
第 30 页
70 字节
x
B
完整 ( 或bf16fp16)
32
120GB
240GB
600GB
1200GB
18x
国标
完整 (pure_bf16)
16
60GB
120GB
300GB
600GB
8x
国标
冻结/LoRA/GaLore/APOLLO/BAdam
16
16GB
32GB
64GB
160GB
2x
国标
QLoRA
8
10GB
20GB
40GB
80GB
x
国标
QLoRA
4
6GB
12GB
24GB
48GB
x/2
国标
QLoRA
2
4GB
8GB
16GB
24GB
x/4
国标

环境搭建

建议使用uv,这是一个非常快速的 Python 环境管理器。安装uv后,您可以使用以下命令创建一个新的Python环境并安装依赖项,注意这不包含xcodec(音频克隆)功能的依赖:

git clone https://github.com/xming521/WeClone.git
cd WeClone
uv venv .venv --python=3.9
source .venv/bin/activate
uv pip install --group main -e .

注意

训练以及推理相关配置统一在文件settings.json

数据准备

请使用PyWxDump提取微信聊天记录。下载软件并解密数据库后,点击聊天备份,导出类型为CSV,可以导出多个联系人或群聊,然后将导出的位于 的 文件夹放在目录即可,也就是不同人聊天记录的文件夹一起放在 。示例数据位于 data/example_chat.csv。wxdump_tmp/exportcsv./data./data/csv

数据预处理

项目默认去除了数据中的手机号、身份证号、邮箱、网址。还提供了一个禁用词词库blocked_words,可以自行添加需要过滤的词句(会默认去掉包括禁用词的整句)。 执行 脚本对数据进行处理。./make_dataset/csv_to_json.py

在同一人连续回答多句的情况下,有三种处理方式:

文件
处理方式
csv_to_json.py
用逗号连接
csv_to_json-单句回答.py(已废弃)
只选择最长的回答作为最终数据
csv_to_json-单句多轮.py
放在了提示词的'history'中

模型下载

首选在Hugging Face下载ChatGLM3 模型。如果您在 Hugging Face 模型的下载中遇到了问题,可以通过下述方法使用魔搭社区,后续训练推理都需要先执行来使用魔搭社区的模型。 由于模型较大,下载过程比较漫长请耐心等待。export USE_MODELSCOPE_HUB=1

export USE_MODELSCOPE_HUB=1 # Windows 使用 `set USE_MODELSCOPE_HUB=1`
git lfs install
git clone https://www.modelscope.cn/ZhipuAI/chatglm3-6b.git

魔搭社区的文件需要更换为 Hugging Face 的modeling_chatglm.py

配置参数并微调模型

  • (可选)修改settings.json选择本地下载好的其他模型。

  • 修改以及来调整显存占用。per_device_train_batch_sizegradient_accumulation_steps

  • 可以根据自己数据集的数量和质量修改、、等参数。num_train_epochslora_ranklora_dropout

单卡训练

运行 进行sft阶段微调,本人loss只降到了3.5左右,降低过多可能会过拟合,我使用了大概2万条整合后的有效数据。src/train_sft.py

python src/train_sft.py
多卡训练
uv pip install deepspeed
deepspeed --num_gpus=使用显卡数量 src/train_sft.py

实战演习

场景1:使用浏览器demo简单推理

python ./src/web_demo.py

场景2:使用接口进行推理

python ./src/api_service.py

场景3:使用常见聊天问题测试

python ./src/api_service.py
python ./src/test_model.py

场景4:部署到聊天机器人

AstrBot方案

AstrBot是易上手的多平台 LLM 聊天机器人及开发框架 ✨ 平台支持 QQ、QQ频道、Telegram、微信、企微、飞书。

使用步骤:

  1. 部署 AstrBot

  2. 在 AstrBot 中部署消息平台

  3. 执行 启动api服务python ./src/api_service.py

  4. 在 AstrBot 中新增服务提供商,类型选择OpenAI,API Base URL根据AstrBot部署方式填写(例如docker部署可能为http://172.17.0.1:8005/v1),模型填写gpt-3.5-turbo

  5. 微调后不支持工具调用,请先关掉默认的工具,消息平台发送指令: ,否则会没有微调后的效果。/tool off reminder

  6. 根据微调时使用的default_system,在 AstrBot 中设置系统提示词。

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

链载AI是专业的生成式人工智能教程平台。提供Stable Diffusion、Midjourney AI绘画教程,Suno AI音乐生成指南,以及Runway、Pika等AI视频制作与动画生成实战案例。从提示词编写到参数调整,手把手助您从入门到精通。
  • 官方手机版

  • 微信公众号

  • 商务合作

  • Powered by Discuz! X3.5 | Copyright © 2025-2025. | 链载Ai
  • 桂ICP备2024021734号 | 营业执照 | |广西笔趣文化传媒有限公司|| QQ