File size: 200 Bytes
b4efe93 | 1 2 3 4 5 6 7 8 9 10 11 |
from hydra.core.config_store import ConfigStore
from trainer.tasks.step_sd_task import StepSDTaskConfig
cs = ConfigStore.instance()
cs.store(group="task", name="step_sd", node=StepSDTaskConfig)
|