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)