跳至主要内容

TensorFlow

此嵌入集成完全在您的浏览器或 Node.js 环境中运行嵌入,使用 TensorFlow.js。这意味着您的数据不会发送到任何第三方,您也不需要注册任何 API 密钥。但是,它确实需要比其他集成更多的内存和处理能力。

npm install @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 文档 以了解更多信息。


此页面对您有帮助吗?


您也可以留下详细的反馈 在 GitHub 上.