CVE 9.8 CRITICAL

CVE-2026-31214_CVE-2026-31214

May 13, 2026 invoker category_cve

9.8 / 10

CRITICAL

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

Description

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.

AI Analysis

Insecure deserialization vulnerability in torch-checkpoint-shrink.py script via PyTorch checkpoint files

Basic Information

ID CVE-2026-31214

Source mitre

Published May 12, 2026 at 00:00

Modified May 13, 2026 at 13:54

Affected Product

Vendor stas00

Product ml-engineering

Version n/a

Affected Versions n/a n/a n/a

CWE Classification

CWE-502

AI Assessment

AI Score 9.8 / 10

AI Severity Critical

Vendor stas00

Product ml-engineering

Version n/a

References

{
    "lastseen": "",
    "description": "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.",
    "published": "2026-05-12T00:00:00.000Z",
    "modified": "2026-05-13T13:54:51.093Z",
    "type": "cve",
    "title": "CVE-2026-31214",
    "source": "mitre",
    "references": "https://github.com/stas00/ml-engineering/blob/master/training/checkpoints/torch-checkpoint-shrink.py#L57\nhttps://www.notion.so/CVE-2026-31214-35d1e1393188813fa40eef73c174cee5",
    "id": "CVE-2026-31214",
    "bulletinFamily": "",
    "cwe": [
        "CWE-502"
    ],
    "cvelist": null,
    "sourceData": "n/a n/a n/a",
    "sourceHref": "",
    "cvss": {
        "score": 9.8,
        "severity": "CRITICAL",
        "vector": "CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H",
        "version": "3.1"
    },
    "cvss2": [],
    "cvss3": {
        "version": "",
        "vectorString": "",
        "baseScore": 0,
        "baseSeverity": "",
        "attackVector": "",
        "attackComplexity": "",
        "privilegesRequired": "",
        "userInteraction": "",
        "scope": "",
        "confidentialityImpact": "",
        "integrityImpact": "",
        "availabilityImpact": "",
        "cvssV3": {
            "version": "",
            "vectorString": "",
            "baseScore": 0,
            "baseSeverity": "",
            "attackVector": "",
            "attackComplexity": "",
            "privilegesRequired": "",
            "userInteraction": "",
            "scope": "",
            "confidentialityImpact": "",
            "integrityImpact": "",
            "availabilityImpact": ""
        }
    },
    "href": "",
    "category_name": "CVE",
    "post_link": "",
    "product": "ml-engineering",
    "version": "n/a",
    "vendor": "stas00",
    "ai_description": "Insecure deserialization vulnerability in torch-checkpoint-shrink.py script via PyTorch checkpoint files",
    "ai_severity": "Critical",
    "ai_vendor": "stas00",
    "ai_product": "ml-engineering",
    "ai_version": "n/a",
    "ai_score": 9.8
}

💭 Join the Security Discussion ❌ Cancel Reply