I saw that this was ran with qm9, do you know how they were able to get a qm9.npz file that works with the model?