复现

本页介绍如何在 LangChain 中运行 Replicate 上的模型。

安装和设置

创建一个 Replicate 账户。获取你的 API 密钥并将其设置为环境变量（REPLICATE_API_TOKEN）。
使用 pip install replicate 安装 Replicate python 客户端。

调用模型

在 Replicate 探索页面上找到一个模型，然后以 owner-name/model-name:version 的格式粘贴模型名称和版本。

例如，对于这个 dolly 模型，点击 API 选项卡。模型名称/版本将是："replicate/dolly-v2-12b:ef0e1aefc61f8e096ebe4db6b2bacc297daf2ef6899f0f7e001ec445893500e5"

只有 model 参数是必需的，但其他任何模型参数也可以通过 input={model_param: value, ...} 的格式传入。

例如，如果我们运行 stable diffusion 并想更改图像尺寸：

Replicate(model="stability-ai/stable-diffusion:db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf", input={'image_dimensions': '512x512'})

请注意，只会返回模型的第一个输出。 接下来，我们可以初始化我们的模型：

llm = Replicate(model="replicate/dolly-v2-12b:ef0e1aefc61f8e096ebe4db6b2bacc297daf2ef6899f0f7e001ec445893500e5")

并运行它：

prompt = """
请逐步推理，用是或否回答以下问题。
狗能开车吗？
"""
llm(prompt)

我们可以使用此语法调用任何 Replicate 模型（不仅仅是 LLM）。例如，我们可以调用 Stable Diffusion：

text2image = Replicate(model="stability-ai/stable-diffusion:db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf", input={'image_dimensions':'512x512'})

image_output = text2image("一只皮卡索风格的骑着摩托车的猫")

安装和设置​

调用模型​

安装和设置

调用模型