配置
用于使用模型配置的辅助模块。有关更多信息,请参阅相应的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
:字符串
|null
.is_encoder_decoder
:布尔值
.max_position_embeddings
:数字
- 静态
.from_pretrained(pretrained_model_name_or_path, options)
⇒Promise.<PretrainedConfig>
- .AutoConfig
.getKeyValueShapes(config)
⇒Record.<字符串, 数组<数字>>
~decoderFeeds
:Record.<字符串, 数组<数字>>
- .PretrainedConfig
- 内部
~loadConfig(pretrained_model_name_or_path, options)
⇒Promise.<对象>
~getNormalizedConfig(config)
⇒对象
~PretrainedOptions
:*
- 静态
configs.PretrainedConfig
所有配置类的基类。有关更多信息,请参阅相应的Python 文档。
种类:configs
的静态类
- .PretrainedConfig
new PretrainedConfig(configJSON)
- 实例
.model_type
:字符串
|null
.is_encoder_decoder
:布尔值
.max_position_embeddings
:数字
- 静态
.from_pretrained(pretrained_model_name_or_path, options)
⇒Promise.<PretrainedConfig>
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 | 字符串 | 预训练配置的路径。 |
options | PretrainedOptions | 加载配置的其他选项。 |
configs.AutoConfig
用于使用from_pretrained
函数实例化预训练配置的辅助类。
种类:configs
的静态类
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 | 字符串 | 配置目录的路径。 |
options | PretrainedOptions | 加载配置的其他选项。 |
configs~getNormalizedConfig(config) ⇒ <code> Object </code>
类型:configs
的内部方法
返回值:Object
- 标准化的配置。
参数 | 类型 |
---|---|
config | PretrainedConfig |
configs~PretrainedOptions : <code> * </code>
类型:configs
的内部类型定义
< > 在 GitHub 上更新