1# Coming soon: AutoModel support
2from src.hf_model import HexaModel
3from src.config import HexaConfig
4
5config = HexaConfig() # Defaults to 5B
6model = HexaModel(config)src/train_hf.py script on a multi-GPU cluster (A100/H100 recommended).