Transformers.js 文档

配置

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 实例。

参数类型描述
configJSON对象

配置的 JSON。


pretrainedConfig.model_type : <code> 字符串 </code> | <code> null </code>

种类PretrainedConfig 的实例属性


pretrainedConfig.is_encoder_decoder : <code> 布尔值 </code>

种类PretrainedConfig 的实例属性


pretrainedConfig.max_position_embeddings : <code> 数字 </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_path字符串

预训练配置的路径。

optionsPretrainedOptions

加载配置的其他选项。


configs.AutoConfig

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

种类configs 的静态类


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_path字符串

配置目录的路径。

optionsPretrainedOptions

加载配置的其他选项。


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

类型configs 的内部方法
返回值Object - 标准化的配置。

参数类型
configPretrainedConfig

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

类型configs 的内部类型定义


< > 在 GitHub 上更新