oobabooga-text-generation-webui可能是最好的语言模型启动器（包含手把手安装教程）

原文：oobabooga-text-generation-webui可能是最好的语言模型启动器（包含手把手安装教程） - 哔哩哔哩

引言：

问：oobabooga是什么？

oobabooga-text-generation-webui是一个用于运行类似Chatglm、RWKV-Raven、Vicuna、MOSS、LLaMA、llama.cpp、GPT-J、Pythia、OPT和GALACTICA等大型语言模型的Gradio Web用户界面。它的目标是成为文本生成的AUTOMATIC1111/stable-diffusion-webui。

这里是它的github链接：https://github.com/oobabooga/text-generation-webui

简单来说它就是个语言模型的启动器。

你可以加载几乎所有的主流模型，并在本地运行他们。语言模型能做什么，你就可以体验到什么，处理文本，个人管家，知识顾问，私人教学，这就是一个不联网的弱化版本的chatgpt，但不同于chatgpt的收费与对话审查限制，客制化的本地模型几乎提供了无限可能。(是的你完全可以训练出属于你自己的猫娘，或者体验各种奇怪内容，咳咳咳)，同时你可以轻松的在huggingface找到十几种以上的模型，活跃的环境让小模型的发展迅速，几乎每周都有几个新模型出现，总有一个符合你的要求。

问：oobabooga-text-generation-webui的主要功能有哪些？

webui支持的功能非常多，大部分重点我已经用加粗标注了。
类似stable-diffusion的，用户友好型交互界面。
较为友善的一键安装模式。
模型兼容性极佳，主流模型都可以加载。
一键式模型切换能力，快速在多个语言模型之间切换。
自带人设管理，一键加载
类似于OpenAI的记事本模式交流界面。
用于对话和角色扮演的聊天模式
与Alpaca、Vicuna、Open Assistant、Dolly、Koala、ChatGLM、MOSS、RWKV-Raven、Galactica、StableLM、WizardLM、Baize、Ziya、Chinese-Vicuna、MPT、INCITE、Wizard Mega、KoAlpaca、Vigogne、Bactrian、h2o和OpenBuddy等各种格式兼容的指令模式
包括LLaVA和MiniGPT-4在内的多模态流程
用于GALACTICA的Markdown输出，包括LaTeX渲染
用于GPT-4chan的漂亮HTML输出
高级聊天功能（发送图片，获取带有TTS的音频响应）
非常高效的文本流处理
参数预设丰富
LLaMA模型支持
4-bitGPTQ模型支持
LoRA（加载和训练）
llama.cpp模型支持
RWKV模型支持
8-bit模式
模型层分布GPU、CPU和磁盘
CPU模式
FlexGen
DeepSpeed ZeRO-3
带流式传输和无流式传输的API

有关我制作的koishi插件的介绍，如果不知道koishi是什么的可以去这里查看：

https://koishi.chat/zh-CN/

简单介绍一下oobabooga-testbot插件：

oobabooga-testbot插件是一个基于oobabooga-text-generation-webui语言模型启动器的qq机器人插件，实现了一些基本的指令，让你可以在qq机器人上实现与语言模型的对话，不需要扛着电脑，有手机就ok。支持人设加载，vits语音回复，支持AI补充tag调用插件绘图等功能。

（毕竟谁不想要一个在线猫娘随时陪伴呢）

插件教程贴：https://forum.koishi.xyz/t/topic/2391

实际上本教程之前是在koishi论坛上发布的，考虑到大伙都不太了解oobabooga所以在这里重新制作一个教程贴，帮助大家安装。

目录：

第一部分oobabooga-text-generation-webui

1.oobabooga-text-generation-webui交互页面展示。

2.如何本地搭建（github一键安装，全程没遇到报错的情况）

3.如何本地搭建（github一键安装，依赖安装出现报错的情况）

4.如何本地搭建（github手动分步安装）

5.模型下载

6.oobabooga交互页面大致导览

7.为什么我推荐大家使用oobabooga-text-generation-webui。

第二部分oobabooga-testbot插件使用以及人设文档创建

1.插件安装

2.启用oobabooga的api调用

3.插件使用教程

4.建立自己的人设文档

第一部分oobabooga-text-generation-webui

1.oobabooga-text-generation-webui交互页面展示。

文本交流页面：

文本交流页面

人设创建页面：

人设创建页面

模型加载与选择界面：

模型加载与选择界面

2.如何本地搭建（全程没遇到报错的情况）

github上提供了两种搭建方式：

     一键安装包。

     分步手动安装。

其他途径搭建：

b站大佬 coyude 提供了一个一键解压安装的懒人包，但我并没有尝试过是否兼容插件。

[AI对话]懒人包v1发布 图形化启动界面(oobabooga/text-generation-webui)可用chatglm/rwkv/vicuna_哔哩哔哩_bilibili

专栏地址：

[懒人包]AI对话懒人包发布 解压即用(text-generation-webui)可加载chatglm/rwkv/vicu - 哔哩哔哩

b站大佬 AI斯基摩人 提供了一个详细的视频教程。

https://www.bilibili.com/video/BV1fP41127ra/?

本专栏将着重用细致的文字描述来帮助各位使用github上的一键安装包安装与分步手动搭建。

github一键安装流程

1.首先前往github主页。（需要科学上网，以下内容都是在科学上网的前提下进行的）

地址：https://github.com/oobabooga/text-generation-webui

2.找到如下选项：

可以看到现在一键安装是支持Windows，Linux和MacOS的。

我们以Windows系统为例：

（注：接下来的全程都是在Windows10系统下的）

选择链接下载：

下载的应该是一个压缩包：

解压缩
这里是解压缩后的文件：

注意！请务必将文件放置在全英文路径下，并且不要包涵空格或者其他特殊符号

这里是一个例子：D:\oobabooga\oobabooga_windows

oobabooga本体大约会占用10g左右的磁盘空间，单个语言模型大小在10g左右，也有特别大的14b模型可以到29g以上，所以请确保你的磁盘空间至少有50g左右。加载模型的时候会将数据导入到运存中，所以你的磁盘读取速度越快越好，个人建议放在SSD中，可以显著加快加载速度。

特别注意：

接下来的步骤需要全程科学上网，才能正确安装，如果不会的话请百度，这里不多做赘述。

然后我们双击运行start_windows.bat

它会自动下载miniconda并且安装在当前目录下。

miniconda安装完成：

这里会让你选择你的GPU类型

A：NVIDIA ，N卡选A

B：AMD ，A卡选B

C：Apple M Series，苹果选C

D：没显卡，或者显卡太弱，用CPU跑模型的用户选D（会导致回复非常慢）

我这里本地是RTX3090显卡，所以我选A，输入A，然后回车继续。

然后它就会开始自动下载一堆东西，不需要进行操作。

这一步需要等待较长时间，取决于你的网速，请你耐心等待。

我这里等待了15分钟左右完成。

出现这个表示下载完成，开始解压安装。请继续等待。

出现这个说明开始安装依赖了。请继续等待。

这一步报错的概率非常大，如果出现红色报错，说明你依赖安装出现了问题。绝大部分安装问题都出在这里。详见下方的第二种方式。

出现以下提示：

恭喜你，你已经成功安装了oobabooga了。

现在的目录结构应该是这样的：

我们运行start_windows.bat，来测试启动oobabooga。

由于我们只安装了oobabooga，并没有下载任何模型，所以启动的时候会提示你选择模型下载。

你可以选择一个下载，但速度会比较慢，个人建议还是用idm直接去huggingface找到对应模型下载比较好。

接下来请跳转到下载模型部分

3.如何本地搭建（依赖安装出现报错的情况）

看起来你很不幸的遇到了报错，报错可能长得很不一样，但是解决方法都大同小异。

现在将以这种安装报错为例：

这代表依赖未完全下载，出现的原因一般是因为你的网络环境不稳定，可以的话请更换节点后再次运行安装程序，如果不行的话就请按照以下内容进行。

再次安装依然失败的情况：

如果依然失败请先关闭当前控制台，然后随便进入一个外国网址确定你的梯子没有问题后，再进行以下步骤。

进入根目录：

依赖安装错误这个问题一般发生在安装完成miniconda，并且成功git clone了github项目之后，这时你的miniconda一般是没有问题的，你的更目录应该和图片一样，存在一个叫做text-generation-webui的文件夹。

如果没有请你不要继续并前往私信或评论区提问。

确定文件存在后你可以继续操作了：

为了解决这个问题我们需要首先进入miniconda。

双击运行cmd_windows.bat文件

这时会弹出一个黑色的窗口:

输入：cd text-generation-webui

回车

输入: pip install -r requirements.txt

回车

这时程序应该开始安装了：

安装完成后oobabooga应该就可以正常运行了。

如果更加不幸的事情发生到你身上了，在安装过程中依然报错，类似这样：

请你继续以下步骤：

现在请你到更目录，进入text-generation-webui，找到requirements文件：

打开后，删除z这三行，并保存退出：

然后回到根目录，找到cmd_windows.bat，双击运行重新创建一个新的控制台：

依次输入并回车

pip install peft

pip install transformers

pip install accelerate

cd text-generation-webui

pip install -r requirements.txt

然后等待安装完成。

完成后，请回到根目录。

运行update_windows.bat来检查一遍所有依赖

如果正常完成：

那么你的oobabooga应该就可以正常工作了。

接下来请跳转到下载模型部分

如果还未解决问题，那么就请你私信up或者在评论区留言

4.如何本地搭建（github手动分步安装）

接下来我将会讲解github上分步手动搭建流程。

手动安装需要一些进阶的知识，所以我是假设你已经具备了这些知识的，所以教程会比较简单。

首先你需要安装Conda或者WSL

这里的话我就不具体讲解如何安装了，b站教程很多，大家自己学习一下就行。

以下我将会以windows系统，安装了conda虚拟环境的方式继续

进入你想要的安装oobabooga的目录

鼠标选择后编辑：

回车，你应该打开了一个cmd窗口：

首先创建虚拟环境：

conda create -n textgen python=3.10.9

激活conda：

conda activate textgen

然后根据你的系统，安装Pytorch

windows环境就是第一句:

pip install torch torchvision torchaudio

然后安装oobabooga

git clone https://github.com/oobabooga/text-generation-webui

cd进入文件夹

cd text-generation-webui

安装依赖

pip install -r requirements.txt

等待依赖跑完就算安装完成了。

更多细节请前往github页面：https://github.com/oobabooga/text-generation-webui

接下来请跳转到模型下载部分

如果出现报错

请尝试使用上一个方法中的出现报错的处理方式进行处理

如果依然不行可以选择在评论区提问，或者私信up。

5.模型下载

接下来我将简单介绍一下如何下载模型并在oobabooga内启动

我们先前往huggingface官网：https://huggingface.co/

我将以Chatglm模型作为例子。

搜索模型：

选择第一个：https://huggingface.co/THUDM/chatglm-6b

然后我们会来到这个界面：

接下来有两个选择，你可以使用oobabooga自带的下载器，下载模型，或者自己选择文件下载。

使用oobabooga内置下载器

优点：简单，不容易出错，模型文件格式肯定正确。

缺点：速度慢。

方法：我们首先点击这里：

然后我们回到这个界面：

选择L选项 Manually specify a Hugging Face model

然后我们把刚刚复制的内容粘贴进去

回车，它就会开始自动下载了。

下载目录为：\oobabooga_windows\text-generation-webui\models

文件夹里面就是下载的模型了。

这种方式只会出现在你没有任何模型的时候

如果你已经有模型了，然后仍然希望用oobabooga进行模型下载，那么可以转跳到交互页面大致导览的model选项卡介绍。

自己下载模型文件：

优点：速度快

缺点：得手动

方法：我们首先进入model存储目录

\oobabooga_windows\text-generation-webui\models

创建一个新的文件夹，比方说：

然后我们去到Chatglm的huggingface页面：

https://huggingface.co/THUDM/chatglm-6b

选择files：

把这里所有的文件都下载到刚刚创建的文件里面：

完成后的模型文件夹内部应该类似这样：

注意！

不同类型的模型有不同的模型格式，一般来说只要把huggingface里所有的文件都下载下来就行了。如果还有问题请加群询问。

接下来我们继续oobabooga的模型加载

你应该已经下载好了模型。

如果model文件夹下只有一个模型，那oobabooga会自动加载，如果有多个模型，像我这样，它就会跳出模型选择的页面：

现在我们以Vicuna-13b作为案例，来启动模型。

（为何不用Chatglm做例子是因为Chatglm需要额外安装一下icetk，稍后我会详细讲的）

（注意！请关闭clash，oobabooga与clash会冲突，导致页面可以启动但是无法与模型对话）

弹出

Running on local URL: http://127.0.0.1:7860

To create a public link, set share=True in launch().

就说明启动成功，然后和stable-diffusion一样，我们访问127.0.0.1:7860

就可以看到webui界面了：

接上面的oobabooga中使用Chatglm模型

我们再加载Chatglm模型的时候，会遇到缺少依赖的报错，一般来说是icetk：

（暂时没找到对应的图片）

为了解决这个问题，我们需要安装icetk。

首先我们来到oobabooga文件夹：

选择cmd_windows.bat运行

输入pip install icetk

回车运行

等待安装完成（如果报错可能是你没开科学上网）

然后返回oobabooga文件夹，找到webui.py文件：

右键用记事本打开：

向下翻到140-180行左右，找到“def run_model()”

在python sever.py后面

添加--trust-remote-code

保存并退出

然后重开oobabooga，你就应该可以加载Chatglm模型了。

如果遇到更多问题，请在评论区提问，或者私信up。

6.oobabooga交互页面大致导览

我会简单介绍一下oobabooga的界面选项，以及如何与模型对话和切换模型。

在页面顶端，我们可以看到这几个选项卡。

第一个是我们与模型对话的主要界面

第二个是人设/角色加载和设定界面

第三个是模型详细参数调节

第四个是模型切换，加载与下载界面

第五个是lora训练界面

第六个是webui的选项卡

我主要来介绍第一个与模型对话的主要界面和第四个模型切换，加载与下载界面

详细设定我就不讲了，大伙自己用翻译看一下就行。

Text generation界面

我们在input输入我们想要与模型对话的内容，然后使用黄色的generate按钮，就可以与模型对话了。下面的几个按钮，就是一些正常的聊天用的按钮，自己摸索一下就都明白了。

然后是这部分，这你有三个模式，基本上就是切换不同的对话框模式，这里展示的chat模式，选择了人设后会出现相对应的头像，更增加代入感：

然后是下面的chat style

这里个选项取决于你的模型支持什么样的chat style，一般来说oobabooga会自动帮你选择最佳的。

最下面是gallery

这个是用来加载角色的。oobabooga自带了一个example角色：

选中角色卡就可以加载。

与模型对话的主要界面介绍就到这里。

接下来是第四个模型切换，加载与下载界面

model界面

我们选择选项卡来到model：

我将会分区来进行详细说明：

左上角是模型选择界面，我们可以直接选择模型进行加载，右边是lora界面，同样可以选择加载，下面是Transformers模型的选项，可以选择限制模型占用的内存大小，右侧可以选择使用的模式：

这里是GPTQ模型使用的选项，一般来说oobabooga会自动选择合适的，但使用较新的模型的时候，你可能得自己选择对应的模式：

这部分是是否自动加载模型，就是你上面的选项选择模型后是否会自动加载。
下面是模型自动下载，和之前提到的oobabooga自带的下载器使用方式一样

只需要在下面的框中输入huggingface上的名称，然后点击download就可以自动下载模型：

这里是llama模型的选项卡：

这里是应用lora，加载模型，重载模型，保存模型设定的按钮：

对于oobabooga界面的介绍就到此为止了

除了我介绍的部分之外，oobabooga还提供非常多的各种类型的功能，同时和stable-diffusion一样，也支持插件安装，大家可以自行探索。

也可以在评论区留言或私信up讨论。

7.为什么我推荐大家使用oobabooga-text-generation-webui

这部分主要是我的主观想法，大伙就当做安利就行了。

我个人对于语言模型非常感兴趣，（主要是因为想要一个个人助理），从openai发布chatgpt开始我就开始广泛的关注小模型。然而初期各种模型对于我这种不太会编程的人而言门槛太高了，很多模型我都跑不起来。

而在大量关注小模型的时候，我注意到国外的小模型发展速度非常快，从RWKV，alpaca，Vicuna，到wizard，stableVicuna，小模型的迭代速度非常的快，几乎每隔几天就有一个新的模型出现，我很希望更多人能够参与到这场变革中来，oobabooga-text-generation-webui就是一个非常好的抓手。

oobabooga-text-generation-webui作为语言模型启动器，对于模型的支持范围可以说是最广的，同时高效的模型加载方式，很大程度上满足了我对于大量模型测试的需求。同时，自带的人设加载功能，也很好满足了我对于个性化的需求。

如果你也和我一样，对于小模型有很浓厚的兴趣，而且希望在模型发布的时候就能第一时间玩到，那我强烈推荐你使用。

第二部分插件使用以及人设文档创建

1.插件安装

在koishi的插件市场搜索oobabooga

点击添加就可以安装了

在插件配置页面进行配置即可正常使用：

2.oobabooga启用api服务

接下来我将讲解如何打开oobabooga的api选项，让插件可以调用本地的oobabooga。

我们来到oobabooga文件夹下，选中webui.py

向下找到140-180行

找到“def run_model()”

在python sever.py后面

添加--api

然后保存并关闭，正常启动webui

你可以在Interface mode选项卡中看到：

这样api服务就正常启动了，你的插件应该可以通过默认的url地址调用oobabooga了。

2.插件使用教程

截止到5月24日，插件版本号是1.2.0

插件提供了多样化的指令：

我们先来看看配置页面：

apiURL是用来填写你的oobabooga的api地址的。想要开启oobabooga的api，请看楼上。

用户使用逻辑如下：

首先我们先试用oob.list指令查看当前的人设

然后我们使用oob.load指令，加载我们想要使用的人设

在插件文件夹内，将会创建一个新的历史记录

然后我们才可以使用oob与模型进行对话

oob指令是基础指令，通过使用oob指令可以直接与模型对话

历史记录将会实时保存在对应文件内，同时historylimit将会限制上下文长度，具体取决于你的设置。

使用oob.undo指令，可以撤回刚刚的一个提问

使用oob.reset指令，可以重置当前会话的历史记录

使用oob.del指令，可以删除当前的会话，来加载新的人设。

oob.tag指令可以，让Ai帮助补充tag，进行绘图。

oob.check指令是一个检查指令，它会检查是是否已经存在会话了

现在的插件已经支持昵称与@唤醒

4.建立自己的人设文档

请注意，人设文档完全在本地读取和保存，每次插件更新都会导致其清空，请注意自己留存备份。

首先我们来到

koishi-plugin-oobabooga-testbot\lib\characters

文件夹下

使用记事本打开任意一个人设文件，你会发现人设文件的格式都是类似的。

人设文件由两部分组成：

基础人设+示例对话

人设文件实际上是以数组的形式存放的，这里是一个比较好理解的例子：

[“Hello”,“Hello”]

这样就组成了一组对话。

但我们的人设文件还要再稍微复杂一些，你可以看到，我们首先给出了一段人设：

“你是人工智能语言模型，你将会用中文回复，你将作为助手，给出有帮助且友善的回复。你只需要作为助手回复。”

然后在标准的对话例子中增加了开头结尾，这里是一个例子：

[“You: ||Hello||”,“Assistant: ||Hello||”]

这是为了让模型更好的认识到哪些是人设，哪些是示例对话。同时规范模型的返回格式，以方便我们后续的处理。

请注意你的人设文件名称：

这是Aqua人设的内部，你可以看到这里的会话格式变成了：

[“You: ||Hello||”,“Aqua: ||Hello||”]

这是因为插件会自动识别人设文件的名称，让模型扮演角色，这样可以让模型的回复更加符合人设。

现在你已经大致了解了人设文件的格式了，你可以尝试性的创建一个自己的人设了。

进一步讨论请私信up或者在评论区讨论

一些想说的话

各位看官如果能坚持看到这里，那真是太了不起了，能忍受我这么长的没有一点乐趣的教程。好累，一边写论文，一边写插件，一边还在写教程，感觉身体被掏空。但各位看官如果能从我的教程中学到点什么，或者哪怕是获得一点启发，我都已经非常高兴了。希望大家能够愉快的使用oobabooga-text-generation-webui，在语言模型快速发展的现在，大家能够更得上时代的步伐，一起体验到小模型给我们带来的便利。

——smzh敬上