aryadomain
/

all_code_base

Model card Files Files and versions

all_code_base / lrm /lrm_sana /trainer /datasets /__init__.py

aryadomain's picture

Add files using upload-large-folder tool

ef8f3ad verified 4 months ago

History Blame Contribute Delete

225 Bytes

	from hydra.core.config_store import ConfigStore

	from trainer.datasets.step_sana_hf_dataset import StepSanaHFDatasetConfig

	cs = ConfigStore.instance()
	cs.store(group="dataset", name="step_sana", node=StepSanaHFDatasetConfig)