File size: 204 Bytes
533920b | 1 2 3 4 5 6 7 |
from hydra.core.config_store import ConfigStore
from trainer.tasks.step_flux_task import StepFluxTaskConfig
cs = ConfigStore.instance()
cs.store(group="task", name="step_flux", node=StepFluxTaskConfig) |
533920b | 1 2 3 4 5 6 7 |
from hydra.core.config_store import ConfigStore
from trainer.tasks.step_flux_task import StepFluxTaskConfig
cs = ConfigStore.instance()
cs.store(group="task", name="step_flux", node=StepFluxTaskConfig) |