Function initialize

src/MaxText/pyconfig.py:200–210

Initializes the configuration by loading YAML files, and applying CLI, env, and kwarg overrides.

(argv: list[str], **kwargs)

def initialize(argv: list[str], **kwargs) -> HyperParameters:
  """Initializes the configuration by loading YAML files, and applying CLI, env, and kwarg overrides."""
  pydantic_config = initialize_pydantic(argv, **kwargs)
  config = HyperParameters(pydantic_config)

  if config.log_config:
    for k, v in sorted(config.get_keys().items()):
      if k != "hf_access_token":
        logger.info("Config param %s: %s", k, v)

  return config


def initialize_pydantic(argv: list[str], **kwargs) -> MaxTextConfig: