Fix loading checkpoints

[Goreg12345]

With Lightning CLI we can only load checkpoints and execute one of the functions fit, validate, predict and test. However, our most prominent use case requires simply loading in the model weights and this is currently not supported.

[jiito]

I think this is supported now! By #16