自定义使用

使用自定义模型

Transformers.js 默认使用托管的预训练模型和预编译的 WASM 二进制文件，开箱即用。您可以按如下方式进行自定义：

设置

import { env } from '@huggingface/transformers';

// Specify a custom location for models (defaults to '/models/').
env.localModelPath = '/path/to/models/';

// Disable the loading of remote models from the Hugging Face Hub:
env.allowRemoteModels = false;

// Set location of .wasm files. Defaults to use a CDN.
env.backends.onnx.wasm.wasmPaths = '/path/to/files/';

如需查看所有可用设置的完整列表，请参阅API 参考文档。

将您的模型转换为 ONNX 格式

我们建议使用Optimum，只需一条命令即可将 PyTorch 模型转换为 ONNX 格式。有关支持的架构的完整列表，请参阅Optimum 文档。