GPT4All
本页面介绍了如何在 LangChain 中使用 GPT4All 包装器。本教程分为两部分:安装与设置,以及用法示例。
安装与设置
- 使用
pip install gpt4all安装 Python 包 - 下载一个 GPT4All 模型 并将其放置在您想要的目录
在此示例中,我们使用 mistral-7b-openorca.Q4_0.gguf:
mkdir models
wget https://gpt4all.io/models/gguf/mistral-7b-openorca.Q4_0.gguf -O models/mistral-7b-openorca.Q4_0.gguf
用法
GPT4All
要使用 GPT4All 包装器,您需要提供预训练模型文件的路径及其配置。
from langchain_community.llms import GPT4All
# 初始化模型。Callbacks 支持 token 级流式传输
model = GPT4All(model="./models/mistral-7b-openorca.Q4_0.gguf", n_threads=8)
# 生成文本
response = model.invoke("Once upon a time, ")
API Reference:GPT4All
您还可以自定义生成参数,例如 n_predict、temp、top_p、top_k 等。
要流式传输模型的预测结果,请添加一个 CallbackManager。
from langchain_community.llms import GPT4All
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
# 支持许多 CallbackHandlers,例如
# from langchain.callbacks.streamlit import StreamlitCallbackHandler
callbacks = [StreamingStdOutCallbackHandler()]
model = GPT4All(model="./models/mistral-7b-openorca.Q4_0.gguf", n_threads=8)
# 生成文本。Tokens 通过 callback manager 流式传输。
model.invoke("Once upon a time, ", callbacks=callbacks)
模型文件
您可以从 GPT4All 客户端下载模型文件。您可以从 GPT4All 网站下载客户端。
有关此内容的更详细演练,请参阅 此笔记 本