Description
PyTorch Lightning through 2.6.5, fixed in commit d710d68, contains a remote code execution vulnerability in the _load_state function that imports and executes attacker-controlled module names from checkpoint _instantiator hyperparameters. Attackers can craft malicious checkpoint files that bypass weights_only=True protections to execute arbitrary code when LightningModule.load_from_checkpoint is called.
Affected products
Version statuses are supplied by the assigning organization. Consult the original record for detailed ranges and platform conditions.
Lightning-AI pytorch-lightning
Default version status: unaffected
0 to ≤ 2.6.5affectedd710d689510d50e800f53b3cd773cbca20b1f86funaffected
pkg:pypi/lightningReferences
- https://github.com/Lightning-AI/pytorch-lightning/issues/21822 ↗
- https://github.com/Lightning-AI/pytorch-lightning/pull/21832 ↗
- https://github.com/Lightning-AI/pytorch-lightning/commit/d710d689510d50e800f53b3cd773cbca20b1f86f ↗
- https://www.vulncheck.com/advisories/pytorch-lightning-arbitrary-code-execution-via-instantiator-hyperparameter ↗