transformers

transformers

TransformersModel

TransformersModel(*args, config_model: Type[PretrainedConfig] = None, automodel: Type[AutoModel] = AutoModel, **kwargs)

Bases: HuggingFaceModel

config instance-attribute

config: PretrainedConfig = config_model

automodel instance-attribute

automodel = automodel if not isinstance(automodel, str) else getattr(modeling_auto, automodel)