CVE-2026-31214

See a problem?
Please try reporting it to the source first.
Source
https://cve.org/CVERecord?id=CVE-2026-31214
Import Source
https://storage.googleapis.com/cve-osv-conversion/osv-output/CVE-2026-31214.json
JSON Data
https://api.osv.dev/v1/vulns/CVE-2026-31214
Published
2026-05-12T00:00:00Z
Modified
2026-10-28T02:30:42Z
Severity
  • 9.8 (Critical) CVSS_V3 - CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H CVSS Calculator
Summary
[none]
Details

The torch-checkpoint-shrink.py script in the ml-engineering project in commit 0099885db36a8f06556efe1faf552518852cb1e0 (2025-20-27) contains an insecure deserialization vulnerability (CWE-502). The script uses torch.load() to process PyTorch checkpoint files (.pt) without enabling the security-restrictive weights_only=True parameter. This oversight 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 in the context of the user running the script.

Database specific 
{
    "cna_assigner":  "mitre",
    "osv_generated_from":  "https://github.com/CVEProject/cvelistV5/tree/main/cves/2026/31xxx/CVE-2026-31214.json"
}
References

Affected packages

Git / github.com/stas00/ml-engineering

Affected ranges

Type
GIT
Repo
https://github.com/stas00/ml-engineering
Events
Introduced
0 Unknown introduced commit / All previous commits are affected
Last Affected
0099885db36a8f06556efe1faf552518852cb1e0
Database specific
Show details 
{
    "extracted_events":  [
        {
            "range":  [
                {
                    "introduced":  "0"
                },
                {
                    "last_affected":  "0099885db36a8f06556efe1faf552518852cb1e0"
                }
            ],
            "source":  "DESCRIPTION"
        }
    ]
}

Database specific

source 
"https://storage.googleapis.com/cve-osv-conversion/osv-output/CVE-2026-31214.json"