mindXtrain / mindxtrain /train /__init__.py
Gregory-L's picture
fork mindXtrain from GitHub (Professor-Codephreak/mindXtrain@661bd41) as the mindX-specific line
dfb775d verified
Raw History Blame Contribute Delete
467 Bytes
"""Training subpackage.
Re-exports the public API symbols expected by the rest of the codebase and the
test suite (`mindxtrain.train.compile_axolotl_yaml`,
`mindxtrain.train.autotune_overrides_summary`, ...).
"""
from mindxtrain.train.axolotl_compile import autotune_overrides_summary, compile_axolotl_yaml
from mindxtrain.train.dispatch import dispatch_training
__all__ = [
"autotune_overrides_summary",
"compile_axolotl_yaml",
"dispatch_training",
]