1from hybridko.model import HybriKoModel, HybriKoConfig
2
3config = HybriKoConfig.from_yaml("config.yaml")
4model = HybriKoModel(config)
5
6# Load checkpoint
7checkpoint = torch.load("checkpoints/checkpoint_step_XXX.pt")
8model.load_state_dict(checkpoint["model_state_dict"])1@misc{hybridko2024,
2 title={HybriKo: Korean Hybrid LLM},
3 year={2024},
4}