{"dataType": "CVE_RECORD", "cveMetadata": {"state": "PUBLISHED", "cveId": "CVE-2026-31214", "assignerOrgId": "8254265b-2729-46b6-b9e3-3dfca2d5bfca", "assignerShortName": "mitre", "dateUpdated": "2026-05-12T15:04:31.056Z", "dateReserved": "2026-03-09T00:00:00.000Z", "datePublished": "2026-05-12T00:00:00.000Z"}, "containers": {"cna": {"providerMetadata": {"orgId": "8254265b-2729-46b6-b9e3-3dfca2d5bfca", "shortName": "mitre", "dateUpdated": "2026-05-12T15:04:31.056Z"}, "descriptions": [{"lang": "en", "value": "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."}], "affected": [{"vendor": "n/a", "product": "n/a", "versions": [{"version": "n/a", "status": "affected"}]}], "references": [{"url": "https://github.com/stas00/ml-engineering/blob/master/training/checkpoints/torch-checkpoint-shrink.py#L57"}, {"url": "https://www.notion.so/CVE-2026-31214-35d1e1393188813fa40eef73c174cee5"}], "problemTypes": [{"descriptions": [{"type": "text", "lang": "en", "description": "n/a"}]}]}}, "dataVersion": "5.2"}

{}

{"cveTags": [], "descriptions": [{"lang": "en", "value": "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."}], "id": "CVE-2026-31214", "lastModified": "2026-05-12T16:16:13.270", "metrics": {}, "published": "2026-05-12T16:16:13.270", "references": [{"source": "cve@mitre.org", "url": "https://github.com/stas00/ml-engineering/blob/master/training/checkpoints/torch-checkpoint-shrink.py#L57"}, {"source": "cve@mitre.org", "url": "https://www.notion.so/CVE-2026-31214-35d1e1393188813fa40eef73c174cee5"}], "sourceIdentifier": "cve@mitre.org", "vulnStatus": "Received"}

{}

{"created": "2026-05-12T18:00:12.685072+00:00", "title": "Insecure deserialization in torch-checkpoint-shrink script leading to remote code execution", "updated": "2026-05-12T18:00:12.685082+00:00", "vendors": []}