CVE-2026-31218

The _load_model() function in the neural_magic_training.py script of the optimate project in commit a6d302f912b481c94370811af6b11402f51d377f (2024-07-21) is vulnerable to insecure deserialization (CWE-502). When loading a model state dictionary from a state_dict.pt file via torch.load(), the function does not enable the weights_only=True security parameter. This allows the deserialization of arbitrary Python objects through the Pickle module. A remote attacker can exploit this by providing a maliciously crafted state_dict.pt file within a directory specified via the --model argument, leading to arbitrary code execution during the deserialization process on the victim's system.
Configurations

No configuration.

History

15 May 2026, 15:16

Type Values Removed Values Added
CWE CWE-502
CVSS v2 : unknown
v3 : unknown
v2 : unknown
v3 : 8.8

12 May 2026, 16:16

Type Values Removed Values Added
New CVE

Information

Published : 2026-05-12 16:16

Updated : 2026-05-15 15:16


NVD link : CVE-2026-31218

Mitre link : CVE-2026-31218

CVE.ORG link : CVE-2026-31218


JSON object : View

Products Affected

No product.

CWE
CWE-502

Deserialization of Untrusted Data