fix: move check_inputs
to target device if available during to_torchscript
.#20873
Merged
Borda merged 2 commits intoLightning-AI:masterfrom Jun 6, 2025
Merged
check_inputs
to target device if available during to_torchscript
.#20873