from hydra.core.config_store import ConfigStore

from trainer.accelerators.debug_accelerator import DebugAcceleratorConfig
from trainer.accelerators.deepspeed_accelerator import DeepSpeedAcceleratorConfig

ACCELERATOR_GROUP_NAME = "accelerator"

cs = ConfigStore.instance()
cs.store(group=ACCELERATOR_GROUP_NAME, name="deepspeed", node=DeepSpeedAcceleratorConfig)
cs.store(group=ACCELERATOR_GROUP_NAME, name="debug", node=DebugAcceleratorConfig)