maxtext.configs.pyconfig module#
Pydantic-based configuration management for MaxText.
- maxtext.configs.pyconfig.initialize(argv=None, **kwargs)[source]#
Initializes the configuration by loading YAML files, and applying CLI, env, and kwarg overrides.
- Parameters:
argv (list[str] | None)
- Return type:
HyperParameters
- maxtext.configs.pyconfig.initialize_pydantic(argv=None, **kwargs)[source]#
Initializes the configuration by loading YAML files, and applying CLI, env, and kwarg overrides. Returns pydantic MaxTextConfig class whereas initialize returns the og HyperParameters
- Parameters:
argv (list[str] | None)
- Return type: