def METHOD_NAME(self, model_name_and_config):
    model_name, config = model_name_and_config
    model = ModelForDownstreamTasks.from_config(config,
                                                model_name,
                                                task_heads=EXTRACTIVE_HEAD)
    head_name = next(iter(EXTRACTIVE_HEAD))
    model.set_task_head(head_name)
    return config, model