from .registry import model_entrypoints from .registry import is_model def build_language_encoder(config, **kwargs): model_name = config['MODEL']['TEXT']['ARCH'] if not is_model(model_name): raise ValueError(f'Unkown model: {model_name}') return model_entrypoints(model_name)(config, **kwargs)