pytorch-lightning

015bc6ef - Fix `load_from_checkpoint` to return model on correct device (#17308)

Fix `load_from_checkpoint` to return model on correct device (#17308) Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> (cherry picked from commit e1ce887fde0a7d93d2ce56aa6655b30b1670208d)