Might be the training code related with the model https://github.com/mistralai/megablocks-public/tree/pstock/m...