fix

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",
 ]

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"