I am trying to load a model from a certain checkpoint and use it for inference. The checkpoint folder looks like this. How do I load the model in torch from this folder. The resources I could find are for loading from a checkpoint file, not a folder.