diff --git a/pkgs/development/python-modules/mlx-lm/default.nix b/pkgs/development/python-modules/mlx-lm/default.nix index f46a8786a06d..d86518623c1f 100644 --- a/pkgs/development/python-modules/mlx-lm/default.nix +++ b/pkgs/development/python-modules/mlx-lm/default.nix @@ -79,6 +79,12 @@ buildPythonPackage (finalAttrs: { "tests/test_models.py::TestModels::test_gated_delta_masked" ]; + disabledTests = [ + # ValueError: [rope] dims must be positive but got 0 + # Reported upstream: https://github.com/ml-explore/mlx-lm/issues/1089 + "test_all_models" + ]; + meta = { description = "Run LLMs with MLX"; homepage = "https://github.com/ml-explore/mlx-lm"; diff --git a/pkgs/development/python-modules/mlx-vlm/default.nix b/pkgs/development/python-modules/mlx-vlm/default.nix index 9a5f6d6bc133..9319be349917 100644 --- a/pkgs/development/python-modules/mlx-vlm/default.nix +++ b/pkgs/development/python-modules/mlx-vlm/default.nix @@ -9,13 +9,13 @@ # dependencies datasets, fastapi, + miniaudio, mlx, mlx-lm, numpy, opencv-python, pillow, requests, - soundfile, tqdm, transformers, uvicorn, @@ -26,35 +26,32 @@ rich, }: -buildPythonPackage rec { +buildPythonPackage (finalAttrs: { pname = "mlx-vlm"; - version = "0.3.9"; + version = "0.4.2"; pyproject = true; src = fetchFromGitHub { owner = "Blaizzy"; repo = "mlx-vlm"; - tag = "v${version}"; - hash = "sha256-L+llrfFo4C++JZ3GjpZi16wMZNXtKrYh3pxhZ5N1n/4="; + tag = "v${finalAttrs.version}"; + hash = "sha256-GmeMcANmztICfYR9Ca5wQfLOugOlK1mt5j3q616n6TQ="; }; build-system = [ setuptools ]; - pythonRelaxDeps = [ - "opencv-python" - ]; dependencies = [ datasets fastapi + miniaudio mlx mlx-lm numpy opencv-python pillow requests - soundfile tqdm transformers uvicorn @@ -91,11 +88,11 @@ buildPythonPackage rec { meta = { description = "Inference and fine-tuning of Vision Language Models (VLMs) on your Mac using MLX"; homepage = "https://github.com/Blaizzy/mlx-vlm"; - changelog = "https://github.com/Blaizzy/mlx-vlm/releases/tag/${src.tag}"; + changelog = "https://github.com/Blaizzy/mlx-vlm/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ]; platforms = [ "aarch64-darwin" ]; }; -} +}) diff --git a/pkgs/development/python-modules/mlx/default.nix b/pkgs/development/python-modules/mlx/default.nix index 3e72fc401a41..6cdc7cb6e6c9 100644 --- a/pkgs/development/python-modules/mlx/default.nix +++ b/pkgs/development/python-modules/mlx/default.nix @@ -33,6 +33,7 @@ let gguf-tools = fetchFromGitHub { owner = "antirez"; repo = "gguf-tools"; + # Tag from https://github.com/ml-explore/mlx/blob/v0.31.1/mlx/io/CMakeLists.txt#L14 rev = "8fa6eb65236618e28fd7710a0fba565f7faa1848"; hash = "sha256-15FvyPOFqTOr5vdWQoPnZz+mYH919++EtghjozDlnSA="; }; @@ -40,14 +41,14 @@ let in buildPythonPackage (finalAttrs: { pname = "mlx"; - version = "0.30.5"; + version = "0.31.1"; pyproject = true; src = fetchFromGitHub { owner = "ml-explore"; repo = "mlx"; tag = "v${finalAttrs.version}"; - hash = "sha256-SV/3MXt+SuJ69XfLfXycold6KgtXSM7OE0KwMSNw+eE="; + hash = "sha256-PiNk/MdMw9Vpat2KuslBTyaFuK+mJ4UvwJqBnysvvUU="; }; patches = [