Advisory Database
  • Advisories
  • Dependency Scanning
  1. pypi
  2. ›
  3. pytorch-lightning
  4. ›
  5. CVE-2026-31221

CVE-2026-31221: PyTorch Lightning load_from_checkpoint has an insecure checkpoint deserialization

May 12, 2026 (updated May 18, 2026)

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.

References

  • github.com/Lightning-AI/pytorch-lightning
  • github.com/advisories/GHSA-75m9-98v2-hjpm
  • nvd.nist.gov/vuln/detail/CVE-2026-31221
  • www.notion.so/CVE-2026-31221-35d1e1393188815f8db7c4fd08076639

Code Behaviors & Features

Detect and mitigate CVE-2026-31221 with GitLab Dependency Scanning

Secure your software supply chain by verifying that all open source dependencies used in your projects contain no disclosed vulnerabilities. Dependency Scanning using SBOM also detects vulnerabilities from proprietary advisories that this site does not include. Learn more about Dependency Scanning using SBOM →

Affected versions

All versions up to 2.6.0

Solution

Unfortunately, there is no solution available yet.

Impact 7.8 HIGH

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

Learn more about CVSS

Weakness

  • CWE-502: Deserialization of Untrusted Data

Source file

pypi/pytorch-lightning/CVE-2026-31221.yml

Spotted a mistake? Edit the file on GitLab.

  • Site Repo
  • About GitLab
  • Terms
  • Privacy Statement
  • Contact

Page generated Sun, 08 Nov 2026 00:17:16 +0000.