Failed to run the checkpoints

Hi there! I'm trying to reproduce the results with the checkpoints but it seems that the checkpoint does not match the current model version. I found a backup version but the model keys are still not matching. Could you help provide the correct model code?