Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
fix: move `check_inputs` to target device if available during `to_torchscript`. (#20873)
(cherry picked from commit ab7b1181ac0c489826a2383e5855edd9584bd134)
G
GdoongMathew committed
083d62a6f876985048a8966f2a05cce3184e7308
Parent: 368c1a4
Committed by Jirka B <[email protected]>
on 6/19/2025, 10:15:36 AM