File size: 235 Bytes
533920b | 1 2 3 4 5 6 7 8 9 10 | 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)
|