GPT4All

本页面介绍如何在 LangChain 中使用 GPT4All 包装器。教程分为两部分:安装和设置,以及示例中的使用方法。

安装和设置

用法

GPT4All

使用 GPT4All 包装器,您需要提供预训练模型文件的路径以及模型的配置。

from langchain.llms import GPT4All
 
# Instantiate the model. Callbacks support token-wise streaming
model = GPT4All(model="./models/gpt4all-model.bin", n_ctx=512, n_threads=8)
 
# Generate text
response = model("Once upon a time, ")
 

您还可以自定义生成参数,例如 n_predict、temp、top_p、top_k 等。

要流式传输模型的预测结果,请添加 CallbackManager。

from langchain.llms import GPT4All
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
# There are many CallbackHandlers supported, such as
# from langchain.callbacks.streamlit import StreamlitCallbackHandler
 
callbacks = [StreamingStdOutCallbackHandler()]
model = GPT4All(model="./models/gpt4all-model.bin", n_ctx=512, n_threads=8)
 
# Generate text. Tokens are streamed through the callback manager.
model("Once upon a time, ", callbacks=callbacks)
 

模型文件

您可以在 pyllamacpp (opens in a new tab) 存储库中找到模型文件下载链接。

有关更详细的演示,请参见 此教程