File size: 311 Bytes
533920b | 1 2 3 4 5 6 7 8 | 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) |