def METHOD_NAME(model_name: str) -> BaseModelParamsT:
  """Retrieves a model config from the global registry."""
  model_imports.import_params(model_name)
  model_class = model_registry.METHOD_NAME(model_name)
  if model_class is None:
    raise ValueError(f'Could not find model `{model_name}`.')
  return model_class