File size: 339 Bytes
c5a246d | 1 2 3 4 5 6 7 8 9 10 11 | 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}")
|