Skip to content

worm128/AI-YinMei-backup

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

48 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AI-YinMei

  • AI 虚拟主播 Vtuber 研发(N 卡版本)
  • AI 名称:吟美
  • 开发者:Winlone
  • B 站频道:程序猿的退休生活
  • Q 群:27831318
  • 版本:1.0
  • 详细笔记: 现在发现有道云笔记网页版本不能查看笔记图片,需要完整教案请进入 Q 群 27831318 获取 pdf 文档
    https://note.youdao.com/s/1k0x7BLt

介绍

  • 支持本地 LLM 模型 chatglm-6b 的 1 代~3 代的 AI 自然语言回复
  • 支持对接 bilibili 直播间弹幕回复和进入直播间欢迎语
  • 支持微软 edge-tts 语音合成
  • 支持聊天记忆模式和扮演卡,可以多角色切换
  • 支持 AI 训练 LLaMA-Factory
  • 支持表情控制 Vtuber Studio
  • 支持绘画 stable-diffusion-webui 输出 OBS 直播间
  • 支持绘画图片鉴黄 public-NSFW-y-distinguish
  • 支持搜索和搜图服务 duckduckgo(需要魔法上网)
  • 支持搜图服务 baidu 搜图(不需要魔法上网)

运行环境

  • Python 3.11.6

调用类库

  • 对应重要的 py 包
    torch:2.1.0+cu121
    peft:0.6.2
    bilibili-api-python:16.1.1
    edge-tts:6.1.9
    pynput:1.7.6
    APScheduler:3.10.4
    transformers:4.35.2
    websocket-client:1.6.4v
    duckduckgo_search:4.1.1
    pyvirtualcam:0.11.0
    opencv-python:4.8.1.78
    Flask:3.0.0
    Flask-APScheduler:1.13.1
    duckduckgo_search:4.1.1

启动方式

1、(必选)启动应用层,在根目录

#进入虚拟环境
& 盘符:路径/pylib/aivenv/Scripts/Activate.ps1
#安装py包
pip install -r requirements.txt
#启动对接b站直播程序
#一:1.b站直播间 2.api web 3.双开:
#二:0.对接后端text-generation-webui接口   1.当前加载LLM本地模型
#三:输入你的B站直播间编号
python bilibili-live-api.py

修改内容须知:

B 站直播间鉴权(B 站浏览器获取 cookie):sessdata、buvid3
Vtuber Studio 表情 websocket 服务:
ws = websocket.WebSocketApp("ws://127.0.0.1:8001",on_open = on_open)
以下是表情鉴权,详细看文档【十三、Vtuber 表情控制-获取令牌和授权】:
vtuber_pluginName="自定义插件名称"
vtuber_pluginDeveloper="winlone"
vtuber_authenticationToken="这个令牌从获取令牌接口获取"
唱歌服务 Auto-Convert-Music 地址:singUrl = "192.168.2.58:1717"
绘画服务 stable-diffusion-webui 地址:drawUrl = "192.168.2.58:7860"
聊天服务 text-generation-webui 地址:tgwUrl = "192.168.2.58:5000"
LLM 模型路径:model_path = "ChatGLM2/THUDM/chatglm2-6b"
LLM 训练模型路径:checkpoint_path = ("LLaMA-Factory/saves/ChatGLM2-6B-Chat/lora/yinmei-20231123-ok-last")
搜索服务代理:duckduckgo_proxies="socks5://127.0.0.1:10806"
搜图服务代理:proxies = {"http": "socks5://127.0.0.1:10806", "https": "socks5://127.0.0.1:10806"}

2、(可选)启动 LLM 聊天服务 text-generation-webui
项目 github:https://github.com/oobabooga/text-generation-webui

#进入虚拟环境
& 盘符:py虚拟空间路径/Scripts/Activate.ps1
#安装py包
pip install -r requirements.txt
#启动text-generation-webui程序,start.bat是我自定义的window启动脚本
./start.bat

3、(可选)启动绘画服务 stable-diffusion-webui
项目 github:https://github.com/AUTOMATIC1111/stable-diffusion-webui

#进入虚拟环境
& 盘符:py虚拟空间路径/Scripts/Activate.ps1
#安装py包
pip install -r requirements.txt
#配置api服务webui-user.bat
@echo off
set PYTHON=.\pydraw\Scripts\python.exe
set GIT=
set VENV_DIR=.\pydraw\
set COMMANDLINE_ARGS=--api
call webui.bat
#启动text-generation-webui程序,start.bat是我自定义的window启动脚本
./webui-user.bat

4、(可选)启动绘画鉴黄服务 public-NSFW-y-distinguish
项目 github:https://github.com/fd-freedom/public-NSFW-y-distinguish

运行环境(必要):Python 3.6.13
pip install -r requirements.txt
# 此文件为本人特制
py nsfw_web.py

5、(可选)启动唱歌服务 Auto-Convert-Music
服务暂不开源

6、(必选)皮肤启动,安装 steam,安装 VTube Studio
这个自行下载 steam 平台,在平台里面有一个 VTube Studio 软件,它就是启动 live2D 的虚拟主播皮肤

7、(必选)其他
安装虚拟声卡:虚拟声卡驱动(Virtual Audio Cable)4.66 官方版

此外,需要在 text-generation-webui/models 路径放入 LLM 模型,我这里放的是 chatgml2 的模型,大家可以任意选择底层 LLM 模型,例如,千问、百川、chatglm、llama 等
更多详细技术细节,请看技术文档:https://note.youdao.com/s/1k0x7BLt

目录说明

  • text-generation-webui【第三方工具】:
    LLM 聚合接口,可以放置 chatglm 等大语言模型,然后进行参数配置后,再输入角色卡进行角色扮演聊天
    https://github.com/oobabooga/text-generation-webui
  • LLaMA-Factory【AI 训练】:
    AI 聚合训练工具,可以界面化配置训练参数,可视化 ai 训练,相当强大
    https://github.com/hiyouga/LLaMA-Factory
  • ChatGLM、ChatGLM2、ChatGLM3【语言模型】:
    放置的是清华大学研发的自然语言模型,可以自行添加如:百川、千问、LLAMA 等其他大语言模型
  • SillyTavern【第三方工具】:
    酒馆,强大的 AI 角色扮演,但是该项目没有公开接口调用,而且 TTS 语言合成很缓慢,暂未集成使用
    https://github.com/SillyTavern/SillyTavern
  • output【输出路径】:
    输出的文本 txt、语音 mp3 文件都在这里
    3
  • ChatGLM2\ptuning【AI 训练】:
    ChatGLM 官方训练例子
  • ChatGLM2\ptuning\zero_nlp【AI 训练】:
    ai 的 lora 训练模式

特别鸣谢

更多关注

  • 讨论 Q 群:27831318