Optimum 文档

标准化配置

您正在查看 主分支 版本,需要从源代码安装. 如果你想使用 pip 进行常规安装,请查看最新的稳定版本 (v1.23.1).
Hugging Face's logo
加入 Hugging Face 社区

并获得增强文档体验

开始使用

标准化配置

🤗 Transformers 中的模型配置类没有标准化。尽管 Transformers 实现了 attribute_map 属性来在一定程度上缓解这个问题,但它并没有让代码中对常见配置属性的推理变得容易。 NormalizedConfig 类试图通过允许以标准化方式访问它们包装的配置属性来解决这个问题。

基类

虽然可以为常见用例创建 NormalizedConfig 子类,但也可以使用 with_args() 类方法直接覆盖 原始属性名 -> 标准化属性名 映射。

class optimum.utils.NormalizedConfig

< >

( config: typing.Union[transformers.configuration_utils.PretrainedConfig, typing.Dict] allow_new: bool = False **kwargs )

参数

  • config (PretrainedConfig) — 要标准化的配置。

处理 PretrainedConfig 属性名的标准化,允许以通用方式访问属性。

现有的标准化配置

class optimum.utils.NormalizedTextConfig

< >

( config: typing.Union[transformers.configuration_utils.PretrainedConfig, typing.Dict] allow_new: bool = False **kwargs )

class optimum.utils.NormalizedSeq2SeqConfig

< >

( config: typing.Union[transformers.configuration_utils.PretrainedConfig, typing.Dict] allow_new: bool = False **kwargs )

class optimum.utils.NormalizedVisionConfig

< >

( config: typing.Union[transformers.configuration_utils.PretrainedConfig, typing.Dict] allow_new: bool = False **kwargs )

optimum.utils.NormalizedTextAndVisionConfig

< >

( config: typing.Union[transformers.configuration_utils.PretrainedConfig, typing.Dict] allow_new: bool = False **kwargs )

< > 在 GitHub 上更新