from hydra.core.config_store import ConfigStore

from trainer.configs.configs import TrainerConfig
from trainer.configs.step_flux_configs import StepFluxTrainerConfig

cs = ConfigStore.instance()
cs.store(name="base_config", node=TrainerConfig)
cs.store(name="step_flux_base_config", node=StepFluxTrainerConfig)