Transformers.js 文档

配置

您正在查看的是需要从源码安装。如果您想使用常规的 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_path字符串

预训练配置的路径。

选项PretrainedOptions

加载配置的附加选项。


configs.AutoConfig

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

类型: configs 的静态类


new AutoConfig()

示例

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

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

类型: AutoConfig 的静态方法


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

类型: configs 的静态方法

参数量类型
configPretrainedConfig

getCacheShapes~cache_values : <code> Record. < string, Array < number > > </code>

类型: getCacheShapes 的内部常量


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

从指定路径加载配置。

类型: configs 的内部方法
返回: Promise.<Object> - 一个解析为已加载配置信息的 Promise。

参数量类型描述
pretrained_model_name_or_path字符串

配置目录的路径。

选项PretrainedOptions

加载配置的附加选项。


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

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

参数量类型
configPretrainedConfig

configs~getKeyValueShapes() : <code> * </code>

类型: configs 的内部方法


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

类型: getKeyValueShapes 的内部常量


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

类型: configs 的内部类型定义


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

类型: configs 的内部类型定义


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

类型: configs 的内部类型定义


< > 在 GitHub 上更新