I'm trying to load the provided model checkpoint using GNN_LP.load_from_checkpoint() as shown in the codebase. However, I encounter the following error during load_from_checkpoint() specifically from the internal call load_state_dict()

I also attempted to clone the repo and run the example code directly using the default model checkpoints, but the same error occurred. Is this a mismatch in model architecture and model checkpoints? Let me know if more context would be helpful. Thank you in advance.