Arbitrary Code Injection Affecting flash-attn package, versions [0,]


Severity

Recommended
0.0
medium
0
10

CVSS assessment by Snyk's Security Team. Learn more

Threat Intelligence

EPSS
0.22% (13th percentile)

Do your applications use this vulnerable package?

In a few clicks we can analyze your entire application and see what components are vulnerable in your application, and suggest you quick fixes.

Test your applications

Snyk Learn

Learn about Arbitrary Code Injection vulnerabilities in an interactive lesson.

Start learning
  • Snyk IDSNYK-PYTHON-FLASHATTN-17670564
  • published29 Jun 2026
  • disclosed11 May 2026
  • creditUnknown

Introduced: 11 May 2026

CVE-2026-31253  (opens in a new tab)
CWE-94  (opens in a new tab)

How to fix?

There is no fixed version for flash-attn.

Overview

flash-attn is a Flash Attention: Fast and Memory-Efficient Exact Attention

Affected versions of this package are vulnerable to Arbitrary Code Injection via the load_checkpoint function and checkpoint loading process. An attacker can achieve arbitrary code execution by supplying a maliciously crafted checkpoint file that is deserialized when loaded during model warmstarting or evaluation.

References

CVSS Base Scores

version 4.0
version 3.1