Something went wrong. Try again.
lol
Something went wrong. Try again.
1.8 kB · 97 lines
Nix
at master
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798{ lib, buildPythonPackage, fetchFromGitHub,
# build-system setuptools,
# dependencies datasets, fastapi, miniaudio, mlx, mlx-lm, numpy, opencv-python, pillow, requests, tqdm, transformers, uvicorn,
# tests psutil, pytestCheckHook, rich, sentencepiece,}:
buildPythonPackage (finalAttrs: { pname = "mlx-vlm"; version = "0.4.4"; pyproject = true;
src = fetchFromGitHub { owner = "Blaizzy"; repo = "mlx-vlm"; tag = "v${finalAttrs.version}"; hash = "sha256-08cSwN8IkERxUaXyT9qAg9vmLw7FvU5qDygAkDsOxpU="; };
build-system = [ setuptools ];
dependencies = [ datasets fastapi miniaudio mlx mlx-lm numpy opencv-python pillow requests tqdm transformers uvicorn ];
pythonImportsCheck = [ "mlx_vlm" ];
nativeCheckInputs = [ psutil pytestCheckHook rich sentencepiece ];
disabledTests = [ # Fatal Python error: Aborted # mlx_vlm/models/multi_modality/vision.py", line 174 in __call__ "test_multi_modality"
# RuntimeError: [metal_kernel] No GPU back-end "test_glm4v" "test_glm4v_moe" "test_kimi_vl" ];
disabledTestPaths = [ # ImportError: cannot import name 'get_class_predicate' from 'mlx_vlm.utils' # This function is indeed not exposed by `mlx_vlm.utils` "mlx_vlm/tests/test_utils.py"
# fixture 'model_path' not found "mlx_vlm/tests/test_smoke.py" ];
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/${finalAttrs.src.tag}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ]; };})