Skip to content

Commit 28bf52a

Browse files
authored
Adding min dependencies for Granite 4 / Mamba support (#14)
Signed-off-by: Mustafa Eyceoz <[email protected]>
1 parent fc2175d commit 28bf52a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ dependencies = [
1717
"instructlab-training>=0.12.0",
1818
"rhai-innovation-mini-trainer>=0.2.0",
1919
"torch>=2.6.0",
20-
"transformers>=4.55.0",
20+
"transformers>=4.57.0",
2121
"numba>=0.50",
2222
"datasets>=4.0.0",
2323
"numpy>=1.26.4,<2.3",
@@ -59,6 +59,7 @@ cuda = [
5959
"kernels>=0.9.0",
6060
"bitsandbytes>=0.47.0",
6161
"liger-kernel>=0.5.10",
62+
"mamba-ssm[causal-conv1d]>=2.2.5",
6263
]
6364

6465
dev = [

0 commit comments

Comments
 (0)