Use custom mlx branch

This commit is contained in:
Ryuichi Leo Takashige
2026-01-26 12:05:28 +00:00
parent 281aaeb013
commit e0d79cdff6
+1 -1
View File
@@ -17,7 +17,7 @@ dependencies = [
"loguru>=0.7.3",
"exo_pyo3_bindings", # rust bindings
"anyio==4.11.0",
"mlx @ git+https://github.com/rltakashige/mlx-jaccl-fix-small-recv.git; sys_platform == 'darwin'",
"mlx @ git+https://github.com/rltakashige/mlx-jaccl-fix-small-recv.git@leo/fix-jaccl-ip; sys_platform == 'darwin'",
"mlx[cpu]==0.30.3; sys_platform == 'linux'",
"mlx-lm @ git+https://github.com/AlexCheema/[email protected]",
"tiktoken>=0.12.0", # required for kimi k2 tokenizer