GitHub Advisoryadvisory · PyPI

GHSA-pq2f-x424-6fjm mamba language model framework vulnerable to insecure deserialization when loading pre-trained models from HuggingFace Hub

CRITICAL Published 2026-05-12 Modified 2026-05-27 Official page ↗
Aliases
CVE-2026-31239
Ecosystem
PyPI
Packages
mamba-ssm
Severity
Critical
CVSS vector
CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H
Weaknesses
CWE-502
Reviewed
2026-05-27

Affected packages

• PyPI mamba-ssm: <= 2.2.6 (no fix yet)

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.

Related

Alias · CVE (1)

  • CVE-2026-31239 The mamba language model framework thru 2.2.6 is vulnerable to insecure deserialization (CWE-502) when loading pre-trained models from… vulnerability

Weakness · CWE (1)

  • CWE-502 Deserialization of Untrusted Data weakness

References