InternLM/internlm/initialize/__init__.py

16 lines
328 B
Python

from .initialize_trainer import initialize_trainer
from .launch import (
get_default_parser,
initialize_distributed_env,
launch_from_slurm,
launch_from_torch,
)
__all__ = [
"get_default_parser",
"initialize_trainer",
"launch_from_slurm",
"launch_from_torch",
"initialize_distributed_env",
]