Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
Expose `weights_only` argument for loading checkpoints in `Fabric.load()` and `Fabric.load_raw()` (#21470)
* Expose weights_only in Fabric.load * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Extend Fabric.load docstring with weights_only argument * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Revert "Merge branch 'fabric-load-weights-only' of https://github.com/CodeVishal-17/pytorch-lightning into fabric-load-weights-only" This reverts commit 6143e653e4b64cbdbc2e95f57d61394b39ad3773, reversing changes made to debde18e0771006c4151b990853e7165265dfbad. * Update fabric.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * add tests * add chlog * add weights_only option to load_raw method for checkpoint loading * add docs for weights_only parameter to Strategy class for checkpoint loading * update docsstrings --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Bhimraj Yadav <[email protected]>
V
Vishal Goyal committed
75dae9f2bb2594ef8b651f3a815cc42f395049c3
Parent: ff1dcd8
Committed by Luca Antiga <[email protected]>
on 1/30/2026, 1:51:21 PM