diff --git a/pyproject.toml b/pyproject.toml
index 1a9fbd0cb..1cc677cf3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -106,8 +106,9 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-# Specific hardware acceleration
+# Flash attention context parallelism
 ring-flash-attn = [
+    "flash-attn==2.8.3 ; sys_platform == 'linux'",
     "ring-flash-attn>=0.1.7",
     "yunchang==0.6.0",
 ]
diff --git a/src/axolotl/_version.py b/src/axolotl/_version.py
new file mode 100644
index 000000000..285c538c4
--- /dev/null
+++ b/src/axolotl/_version.py
@@ -0,0 +1,33 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple, Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = "0.12.1.dev87+g031a2474c.d20250911"
+__version_tuple__ = version_tuple = (0, 12, 1, "dev87", "g031a2474c.d20250911")
+
+__commit_id__ = commit_id = "g031a2474c"