Downgrade Pytorch Lightning, I'm assuming this may have been missed due to the focus on v1.