from hydra.core.config_store import ConfigStore from trainer.tasks.step_sana_task import StepSanaTaskConfig cs = ConfigStore.instance() cs.store(group="task", name="step_sana", node=StepSanaTaskConfig)