OLMo
OLMo copied to clipboard
Add script for comparing model state
This PR adds a script for comparing model state of 2 different OLMo models/checkpoints. This helped reveal that one of the LUMI containers was not loading/saving checkpoints correctly.