Skip to content
OpenAdvisories
Critical · 9.8 CVE-2026-31239 EUVD-2026-29562GHSA-pq2f-x424-6fjmPYSEC-2026-406

mamba language model framework vulnerable to insecure deserialization when loading pre-trained models from HuggingFace Hub

Published May 12, 2026 · Updated May 14, 2026 · Assigned by mitre

Description

The mamba language model framework thru 2.2.6 is vulnerable to insecure deserialization (CWE-502) when loading pre-trained models from HuggingFace Hub. The MambaLMHeadModel.from_pretrained() method uses torch.load() to load the pytorch_model.bin weight file without enabling the security-restrictive weights_only=True parameter. This allows the deserialization of arbitrary Python objects via the pickle module. An attacker can exploit this by publishing a malicious model repository on HuggingFace Hub. When a victim loads a model from this repository, arbitrary code is executed on the victim's system in the context of the mamba process.

Affected packages

EcosystemPackageAffectedFixedRecord
PyPI mamba-ssm <= 2.2.6 — GHSA-pq2f-x424-6fjm

References