Skip to content
HighVulnerabilityLLM-specific

CVE-2026-31239: mamba language model framework insecure deserialization when loading models

Identifier
CVE-2026-31239
Published
Record updated
View JSON
Affected
  • mamba-ssm <= 2.2.6
Fixed in
No fixed version was stated when the source was last read.
Known exploitation
Not listed in the CISA Known Exploited Vulnerabilities catalog at the last check.
EPSS
0.6%

Summary

The mamba language model framework through 2.2.6 is affected by insecure deserialization (CWE-502) when loading pre-trained models from HuggingFace Hub. The MambaLMHeadModel.from_pretrained() method calls torch.load() on pytorch_model.bin without setting weights_only=True, so arbitrary Python objects can be deserialized via the pickle module. An attacker who publishes a malicious model repository can execute arbitrary code on a victim's system, in the context of the mamba process, when the victim loads a model from it.

Mitigation

The source does not state a fix yet. Check the original advisory for updates.