from hydra.core.config_store import ConfigStore from trainer.models.sana_preference_model import SanaPreferenceModelConfig cs = ConfigStore.instance() cs.store(group="model", name="step_sana_base", node=SanaPreferenceModelConfig)