pytorch-lightning

f4178d5d - test: move the pretrained model to the same device as current model before assertion (#19668)