References to Advisories, Solutions, and Tools

By selecting these links, you will be leaving NIST webspace. We have provided these links to other web sites because they may have information that would be of interest to you. No inferences should be drawn on account of other sites being referenced, or not, from this page. There may be other web sites that are more appropriate for your purpose. NIST does not necessarily endorse the views expressed, or concur with the facts presented on these sites. Further, NIST does not endorse any commercial products that may be mentioned on these sites. Please address comments about this page to [email protected].

Known Affected Software Configurations Switch to CPE 2.2

Change History

AV:L/AC:L/PR:N/UI:R/S:U/C:H/I:H/A:H

CWE-502

OR
          *cpe:2.3:a:lightningai:pytorch_lightning:*:*:*:*:*:python:*:* versions up to (including) 2.6.0

MITRE: https://github.com/Lightning-AI/pytorch-lightning Types: Product

MITRE: https://www.notion.so/CVE-2026-31221-35d1e1393188815f8db7c4fd08076639 Types: Exploit, Third Party Advisory

PyTorch-Lightning versions 2.6.0 and earlier contain an insecure deserialization vulnerability (CWE-502) in the checkpoint loading mechanism. The LightningModule.load_from_checkpoint() method, which is commonly used to load saved model states, internally calls torch.load() without setting the security-restrictive weights_only=True parameter. This default behavior allows the deserialization of arbitrary Python objects via the Pickle module. A remote attacker can exploit this by providing a maliciously crafted checkpoint file, leading to arbitrary code execution on the victim's system when the file is loaded.

https://github.com/Lightning-AI/pytorch-lightning

https://www.notion.so/CVE-2026-31221-35d1e1393188815f8db7c4fd08076639