Optimum 文档

标准化配置

您正在查看 main 版本,该版本需要从源码安装。如果您想要常规 pip 安装,请查看最新的稳定版本 (v1.24.0)。
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 )

class optimum.utils.NormalizedTextAndVisionConfig

< >

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

< > 更新 在 GitHub 上