TensorFlow
此嵌入集成完全在您的浏览器或 Node.js 环境中运行嵌入,使用 TensorFlow.js。这意味着您的数据不会发送到任何第三方,您也不需要注册任何 API 密钥。但是,它确实需要比其他集成更多的内存和处理能力。
- npm
- Yarn
- pnpm
npm install @langchain/community @langchain/core @tensorflow/[email protected] @tensorflow/[email protected] @tensorflow-models/[email protected] @tensorflow/tfjs-backend-cpu
yarn add @langchain/community @langchain/core @tensorflow/[email protected] @tensorflow/[email protected] @tensorflow-models/[email protected] @tensorflow/tfjs-backend-cpu
pnpm add @langchain/community @langchain/core @tensorflow/[email protected] @tensorflow/[email protected] @tensorflow-models/[email protected] @tensorflow/tfjs-backend-cpu
import "@tensorflow/tfjs-backend-cpu";
import { TensorFlowEmbeddings } from "langchain/embeddings/tensorflow";
const embeddings = new TensorFlowEmbeddings();
此示例使用 CPU 后端,它在任何 JS 环境中都可以工作。但是,您可以使用 TensorFlow.js 支持的任何后端,包括 GPU 和 WebAssembly,它们会快得多。对于 Node.js,您可以使用 @tensorflow/tfjs-node
包,对于浏览器,您可以使用 @tensorflow/tfjs-backend-webgl
包。请参阅 TensorFlow.js 文档 以了解更多信息。