diff --git a/setup.py b/setup.py index 055842e50..6c970468d 100644 --- a/setup.py +++ b/setup.py @@ -156,7 +156,7 @@ def append_nvcc_threads(nvcc_extra_args): setup( name="flash_attn", - version="0.2.4", + version="0.2.5", packages=find_packages( exclude=("build", "csrc", "include", "tests", "dist", "docs", "benchmarks", "flash_attn.egg-info",) ),