Skip to content
GitHub
View on GitHub

MsSwiftConfig

API reference for MsSwiftConfig

from modal_training_gym.frameworks.ms_swift.config import MsSwiftConfig

Top-level wrapper that composes an ms-swift Megatron SFT run.

FieldTypeDefaultDescription
dataset`‘DatasetConfigNone’`None
model`‘ModelConfigurationNone’`None
wandb`‘WandbConfigNone’`None
framework_configMsSwiftFrameworkConfigMsSwiftFrameworkConfig()ms-swift Megatron training and infrastructure settings. Default MsSwiftFrameworkConfig().

build_app(self, *, name: 'str | None' = None) -> "'App'"

Section titled “build_app(self, *, name: 'str | None' = None) -> "'App'"”

cli_args(self, *, output_dir: 'str') -> 'list[str]'

Section titled “cli_args(self, *, output_dir: 'str') -> 'list[str]'”

Source: modal_training_gym/frameworks/ms_swift/config.py