CVE-2026-31214 Deserialization in n/a
A remote attacker can execute arbitrary code by providing a malicious PyTorch checkpoint file to the torch-checkpoint-shrink.py script due to insecure deserialization.
- Exploited in the wild
- Not known
- Login needed
- Not needed
- Attackable
- Over the internet
- Victim action needed
- Not needed
- Fix
- None listed
What is CVE-2026-31214?
CVE-2026-31214 is a critical unsafe deserialization vulnerability in n/a, published on 12 May 2026. An attacker can exploit it over the internet without logging in, with no action from the victim.
Official 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.
Weakness: CWE-502
Which versions are affected by CVE-2026-31214?
How do I fix CVE-2026-31214?
No fix has been published for CVE-2026-31214 yet. Limit exposure of the affected software until one is available.
Upgrade, and only deserialise data from trusted sources or with an allow-list of types. More about Unsafe deserialization.
Is CVE-2026-31214 being exploited?
There is no known exploitation of CVE-2026-31214 in the wild and no public exploit code.
Sources
Also known as GHSA-qw5f-8jqw-9832.