接入TGUI
This commit is contained in:
@@ -2,7 +2,7 @@
|
||||
|
||||
## 1. 先运行text-generation
|
||||
``` sh
|
||||
# 下载模型
|
||||
# 下载模型( text-generation 这么牛的项目,别忘了给人家star )
|
||||
git clone https://github.com/oobabooga/text-generation-webui.git
|
||||
|
||||
# 安装text-generation的额外依赖
|
||||
@@ -12,28 +12,25 @@ pip install accelerate bitsandbytes flexgen gradio llamacpp markdown numpy peft
|
||||
cd text-generation-webui
|
||||
|
||||
# 下载模型
|
||||
python download-model.py facebook/opt-1.3b
|
||||
|
||||
# 其他可选如 facebook/galactica-1.3b
|
||||
python download-model.py facebook/galactica-1.3b
|
||||
# 其他可选如 facebook/opt-1.3b
|
||||
# facebook/galactica-6.7b
|
||||
# facebook/galactica-120b
|
||||
|
||||
# Pymalion 6B is a proof-of-concept dialogue model based on EleutherAI's GPT-J-6B.
|
||||
# facebook/pygmalion-1.3b
|
||||
# facebook/pygmalion-1.3b 等
|
||||
# 详情见 https://github.com/oobabooga/text-generation-webui
|
||||
|
||||
# 启动text-generation,注意把模型的斜杠改成下划线
|
||||
python server.py --cpu --listen --listen-port 7860 --model facebook_galactica-1.3b
|
||||
```
|
||||
|
||||
## 2. 修改config.py
|
||||
``` sh
|
||||
# LLM_MODEL格式较复杂 TGUI:[模型]@[ws地址]:[ws端口] , 端口要和上面给定的端口一致
|
||||
LLM_MODEL = "TGUI:galactica-1.3b@localhost:7860"
|
||||
```
|
||||
# LLM_MODEL格式为 [模型]@[ws地址] @[ws端口]
|
||||
LLM_MODEL = "pygmalion-1.3b@localhost@7860"
|
||||
```
|
||||
|
||||
|
||||
## 3. 运行!
|
||||
```
|
||||
``` sh
|
||||
cd chatgpt-academic
|
||||
python main.py
|
||||
```
|
||||
|
||||
Reference in New Issue
Block a user