pytorch-lightning

6dfaebab - Avoid deprecated `load_state_dict` for distributed checkpoints in PyTorch 2.2+ (#19298)