pytorch-lightning
Expose `weights_only` argument for loading checkpoints in `Fabric.load()` and `Fabric.load_raw()`
#21470
Merged

Expose `weights_only` argument for loading checkpoints in `Fabric.load()` and `Fabric.load_raw()` #21470

CodeVishal-17
CodeVishal-17 Expose weights_only in Fabric.load
28e32128
CodeVishal-17 CodeVishal-17 requested a review from lantiga lantiga 63 days ago
CodeVishal-17 CodeVishal-17 requested a review from tchaton tchaton 63 days ago
CodeVishal-17 CodeVishal-17 requested a review from justusschock justusschock 63 days ago
CodeVishal-17 CodeVishal-17 requested a review from ethanwharris ethanwharris 63 days ago
github-actions github-actions added fabric
pre-commit-ci[bot] [pre-commit.ci] auto fixes from pre-commit.com hooks
dee13927
GdoongMathew
GdoongMathew requested changes on 2026-01-06
CodeVishal-17 Extend Fabric.load docstring with weights_only argument
debde18e
CodeVishal-17 Merge branch 'fabric-load-weights-only' of https://github.com/CodeVis…
6143e653
pre-commit-ci[bot] [pre-commit.ci] auto fixes from pre-commit.com hooks
e40fa990
GdoongMathew
GdoongMathew commented on 2026-01-06
CodeVishal-17 Revert "Merge branch 'fabric-load-weights-only' of https://github.com…
0dcbc45a
CodeVishal-17 Update fabric.py
4a55b96f
CodeVishal-17 Merge branch 'fabric-load-weights-only' of https://github.com/CodeVis…
2833a3b0
pre-commit-ci[bot] [pre-commit.ci] auto fixes from pre-commit.com hooks
fad0964e
GdoongMathew
GdoongMathew approved these changes on 2026-01-07
codecov
bhimrazy Merge branch 'master' into fabric-load-weights-only
d4175e6b
bhimrazy Merge branch 'master' into fabric-load-weights-only
3b79ebca
bhimrazy add tests
02967dc3
bhimrazy add chlog
26b6df96
bhimrazy add weights_only option to load_raw method for checkpoint loading
12aeac9f
bhimrazy add docs for weights_only parameter to Strategy class for checkpoint …
874ad7e2
bhimrazy update docsstrings
3aee89c3
bhimrazy bhimrazy changed the title Expose weights_only argument in Fabric.load Expose `weights_only` argument for loading checkpoints in `Fabric.load()` and `Fabric.load_raw()` 57 days ago
bhimrazy
bhimrazy approved these changes on 2026-01-12
tchaton
tchaton approved these changes on 2026-01-12
tchaton tchaton merged 20602996 into master 57 days ago
CodeVishal-17 CodeVishal-17 deleted the fabric-load-weights-only branch 45 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone