Commit 886f05ec authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.mlx-lm: skip failing test 

Context: https://github.com/ml-explore/mlx/pull/3230

Reported upstream: https://github.com/ml-explore/mlx-lm/issues/1089
parent b606b2e4

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";