def METHOD_NAME(self, cfg: Config):
    if self.cfg_modify_fn is not None:
        cfg = self.cfg_modify_fn(cfg)
    self.prepare_labels(cfg)
    if not hasattr(cfg.model, 'label2id') and not hasattr(
            cfg.model, 'id2label'):
        if self.id2label is not None:
            cfg.model['id2label'] = self.id2label
        if self.label2id is not None:
            cfg.model['label2id'] = self.label2id
    return cfg