Transformers.js 文档
配置
您正在查看 main 版本,该版本需要从源代码安装。如果您想要常规 npm 安装,请查看最新的稳定版本 (v3.0.0)。
加入 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,
// ...
// }
- 配置
- 静态
- .PretrainedConfig
new PretrainedConfig(configJSON)
- 实例
.model_type
:string
|null
.is_encoder_decoder
:boolean
.max_position_embeddings
:number
- 静态
.from_pretrained(pretrained_model_name_or_path, options)
⇒Promise.<PretrainedConfig>
- .AutoConfig
.getKeyValueShapes(config)
⇒Record.<string, Array<number>>
~decoderFeeds
:Record.<string, Array<number>>
- .PretrainedConfig
- 内部
~loadConfig(pretrained_model_name_or_path, options)
⇒Promise.<Object>
~getNormalizedConfig(config)
⇒Object
~PretrainedOptions
:*
~ProgressCallback
:*
~ProgressInfo
:*
- 静态
configs.PretrainedConfig
所有配置类的基类。有关更多信息,请参阅相应的 Python 文档。
类型:configs
的静态类
- .PretrainedConfig
new PretrainedConfig(configJSON)
- 实例
.model_type
:string
|null
.is_encoder_decoder
:boolean
.max_position_embeddings
:number
- 静态
.from_pretrained(pretrained_model_name_or_path, options)
⇒Promise.<PretrainedConfig>
new PretrainedConfig(configJSON)
创建一个新的 PreTrainedTokenizer 实例。
参数 | 类型 | 描述 |
---|---|---|
configJSON | Object | 配置的 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 | string | 预训练配置的路径。 |
options | 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.getKeyValueShapes(config) ⇒ <code> Record. < string, Array < number > > </code>
类型:configs
的静态方法
参数 | 类型 |
---|---|
config | PretrainedConfig |
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 | string | 配置目录的路径。 |
options | PretrainedOptions | 用于加载配置的附加选项。 |
configs~getNormalizedConfig(config) ⇒ <code> Object </code>
类型:configs
的内部方法
返回:Object
- 规范化的配置。
参数 | 类型 |
---|---|
config | PretrainedConfig |
configs~PretrainedOptions : <code> * </code>
类型:configs
的内部类型定义
configs~ProgressCallback : <code> * </code>
类型:configs
的内部类型定义
configs~ProgressInfo : <code> * </code>
类型:configs
的内部类型定义
< > 在 GitHub 上更新