Skip to main content

Baseten

Baseten 是一个提供完整基础设施的服务商,可用于高效、可扩展和成本效益的部署和提供机器学习模型。

作为一个模型推理平台,Baseten 是 LangChain 生态系统中的一个提供商Baseten 集成目前实现了一个单一的组件,即 LLMs,但计划中还会有更多!

Baseten 让你可以运行开源模型,比如 Llama 2 或 Mistral,并在专用 GPU 上运行专有或经过精细调整的模型。如果你习惯于像 OpenAI 这样的服务商,使用 Baseten 会有一些不同之处:

  • 你不是按 token 支付,而是按 GPU 使用的分钟数支付。

  • Baseten 上的每个模型都使用我们的开源模型打包框架 Truss,以实现最大的可定制性。

  • 虽然我们有一些与 OpenAI ChatCompletions 兼容的模型,你也可以使用 Truss 定义自己的 I/O 规范。

了解更多 关于模型 ID 和部署。

Baseten 文档 中了解更多关于 Baseten 的信息。

安装和设置

要在 LangChain 中使用 Baseten 模型,你需要两样东西:

将你的 API 密钥导出为一个名为 BASETEN_API_KEY 的环境变量。

export BASETEN_API_KEY="在这里粘贴你的 API 密钥"

LLMs

查看 使用示例

from langchain_community.llms import Baseten

Was this page helpful?


You can leave detailed feedback on GitHub.