SIGN IN SIGN UP

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

0 0 0 Python

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