Source code for tango.integrations.transformers.config
from transformers import AutoConfig, PretrainedConfig
from tango.common import Registrable
[docs]class Config(PretrainedConfig, Registrable):
"""
A :class:`~tango.common.Registrable` version of transformers'
:class:`~transformers.PretrainedConfig`.
"""
default_implementation = "auto"
"""
The default registered implementation just calls
:meth:`transformers.AutoConfig.from_pretrained()`.
"""
Config.register("auto", constructor="from_pretrained")(AutoConfig)