File size: 225 Bytes
b4efe93 | 1 2 3 4 5 6 | from hydra.core.config_store import ConfigStore
from trainer.datasets.step_flux_hf_dataset import StepFluxHFDatasetConfig
cs = ConfigStore.instance()
cs.store(group="dataset", name="step_flux", node=StepFluxHFDatasetConfig) |