from transformers import MPNetConfig, MPNetModel


def build_text_encoder(config):
    if config.model_type == "mpnet":
        fields = config.to_dict()
        fields.pop("model_type", None)
        return MPNetModel(MPNetConfig(**fields))
    else:
        raise NotImplementedError(f"Unsupported text model_type: {config.model_type}")