Transformers.js 文档

配置

您正在查看 main 版本,该版本需要从源代码安装。如果您想要常规 npm 安装,请查看最新的稳定版本 (v3.0.0)。
Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

配置

用于使用模型配置的辅助模块。有关更多信息,请参阅相应的 Python 文档

示例: 加载 AutoConfig

import { AutoConfig } from '@huggingface/transformers';
const config = await AutoConfig.from_pretrained('bert-base-uncased');
console.log(config);
// PretrainedConfig {
//   "model_type": "bert",
//   "is_encoder_decoder": false,
//   "architectures": [
//       "BertForMaskedLM"
//   ],
//   "vocab_size": 30522
//   "num_attention_heads": 12,
//   "num_hidden_layers": 12,
//   "hidden_size": 768,
//   "max_position_embeddings": 512,
//   ...
// }

configs.PretrainedConfig

所有配置类的基类。有关更多信息,请参阅相应的 Python 文档

类型configs 的静态类


new PretrainedConfig(configJSON)

创建一个新的 PreTrainedTokenizer 实例。

参数类型描述
configJSONObject

配置的 JSON。


pretrainedConfig.model_type : <code> string </code> | <code> null </code>

类型PretrainedConfig 的实例属性


pretrainedConfig.is_encoder_decoder : <code> boolean </code>

类型PretrainedConfig 的实例属性


pretrainedConfig.max_position_embeddings : <code> number </code>

类型PretrainedConfig 的实例属性


PretrainedConfig.from_pretrained(pretrained_model_name_or_path, options) ⇒ <code> Promise. < PretrainedConfig > </code>

从给定的 pretrained_model_name_or_path 加载预训练配置。

类型PretrainedConfig 的静态方法
返回Promise.<PretrainedConfig> - PretrainedConfig 类的新实例。
抛出:

  • Error 如果在 `pretrained_model_name_or_path` 中未找到 config.json,则抛出错误。
参数类型描述
pretrained_model_name_or_pathstring

预训练配置的路径。

optionsPretrainedOptions

用于加载配置的附加选项。


configs.AutoConfig

辅助类,用于实例化具有 from_pretrained 函数的预训练配置。

类型configs 的静态类


new AutoConfig()

示例

const config = await AutoConfig.from_pretrained('Xenova/bert-base-uncased'); 

AutoConfig.from_pretrained() : <code> * </code>

类型AutoConfig 的静态方法


configs.getKeyValueShapes(config) ⇒ <code> Record. < string, Array < number > > </code>

类型configs 的静态方法

参数类型
configPretrainedConfig

getKeyValueShapes~decoderFeeds : <code> Record. < string, Array < number > > </code>

类型getKeyValueShapes 的内部常量


configs~loadConfig(pretrained_model_name_or_path, options) ⇒ <code> Promise. < Object > </code>

从指定路径加载配置。

类型configs 的内部方法
返回Promise.<Object> - 一个 Promise,它使用有关已加载配置的信息进行解析。

参数类型描述
pretrained_model_name_or_pathstring

配置目录的路径。

optionsPretrainedOptions

用于加载配置的附加选项。


configs~getNormalizedConfig(config) ⇒ <code> Object </code>

类型configs 的内部方法
返回Object - 规范化的配置。

参数类型
configPretrainedConfig

configs~PretrainedOptions : <code> * </code>

类型configs 的内部类型定义


configs~ProgressCallback : <code> * </code>

类型configs 的内部类型定义


configs~ProgressInfo : <code> * </code>

类型configs 的内部类型定义


< > 在 GitHub 上更新