| 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) |