Unverified Commit c3e05a12 authored by Nick Cao's avatar Nick Cao Committed by GitHub
Browse files

python3Packages.mlx: 0.30.5 -> 0.31.1 (#504828)

parents 49ca49be ceba8441

pkgs/development/python-modules/mlx-lm/default.nix

+6 −0
Original line number Diff line number Diff line
@@ -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";

pkgs/development/python-modules/mlx-vlm/default.nix

+8 −11
Original line number Diff line number Diff line
@@ -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"

    ];

  };

}
 
})

pkgs/development/python-modules/mlx/default.nix

+3 −2
Original line number Diff line number Diff line
@@ -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 = [