GitHub Advisoryadvisory · PyPI
GHSA-pq2f-x424-6fjm mamba language model framework vulnerable to insecure deserialization when loading pre-trained models from HuggingFace Hub
CRITICAL
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