talexander/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Releases Activity

python3Packages.mlx-lm: 0.26.0 -> 0.26.3

Browse Source 
Diff: https://github.com/ml-explore/mlx-lm/compare/v0.26.0...v0.26.3

Changelog: https://github.com/ml-explore/mlx-lm/releases/tag/v0.26.3
This commit is contained in:
Gaetan Lepage 2025-08-18 13:27:10 +02:00
parent 3b9bf641c3
commit 4ebe160b24
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pkgs/development/python-modules/mlx-lm/default.nix
View File

@ -2,13 +2,20 @@
lib, lib,
buildPythonPackage, buildPythonPackage,
fetchFromGitHub, fetchFromGitHub,
# build-system
setuptools, setuptools,
# dependencies
jinja2, jinja2,
mlx, mlx,
numpy, numpy,
protobuf, protobuf,
pyyaml, pyyaml,
transformers, transformers,
# tests
lm-eval,
sentencepiece, sentencepiece,
pytestCheckHook, pytestCheckHook,
writableTmpDirAsHomeHook, writableTmpDirAsHomeHook,
@ -40,9 +47,10 @@ buildPythonPackage rec {
]; ];
nativeCheckInputs = [ nativeCheckInputs = [
writableTmpDirAsHomeHook lm-eval
pytestCheckHook pytestCheckHook
sentencepiece sentencepiece
writableTmpDirAsHomeHook
]; ];
pythonImportsCheck = [ pythonImportsCheck = [
@ -62,6 +70,7 @@ buildPythonPackage rec {
"tests/test_prompt_cache.py::TestPromptCache::test_cache_with_generate" "tests/test_prompt_cache.py::TestPromptCache::test_cache_with_generate"
"tests/test_prompt_cache.py::TestPromptCache::test_trim_cache_with_generate" "tests/test_prompt_cache.py::TestPromptCache::test_trim_cache_with_generate"
# RuntimeError: [metal_kernel] No GPU back-end. # RuntimeError: [metal_kernel] No GPU back-end.
"tests/test_losses.py"
"tests/test_models.py::TestModels::test_bitnet" "tests/test_models.py::TestModels::test_bitnet"
]; ];