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. Learn more about Dependency Scanning →

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 Tue, 23 Jun 2026 12:24:45 +0000.