File size: 235 Bytes
18e07e6 | 1 2 3 4 5 6 7 8 9 10 | from hydra.core.config_store import ConfigStore
from trainer.models.flux_preference_model import FluxPreferenceModelConfig
cs = ConfigStore.instance()
cs.store(group="model", name="step_flux_base", node=FluxPreferenceModelConfig)
|