Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110{ lib, buildPythonPackage, fetchFromGitHub, nix-update-script,
# build-system cmake, ninja, scikit-build-core,
# buildInputs ggml,
# dependencies numpy, typing-extensions,
# optional-dependencies accelerate, sentencepiece, torch, torchaudio, torchvision, transformers, cairosvg, mkdocs, mkdocs-material, mkdocstrings, pillow,
# tests pytestCheckHook,}:
buildPythonPackage (finalAttrs: { pname = "ggml-python"; version = "0.0.45"; pyproject = true; __structuredAttrs = true;
src = fetchFromGitHub { owner = "abetlen"; repo = "ggml-python"; tag = "v${finalAttrs.version}"; # ggml-python expects an older version of ggml than pkgs.ggml's fetchSubmodules = true; hash = "sha256-rPbYp6if9bCiQGfM7ZC84hkJKadE2mwC9N3elgVfQBc="; };
build-system = [ cmake ninja scikit-build-core ]; dontUseCmakeConfigure = true;
buildInputs = [ ggml ];
dependencies = [ numpy typing-extensions ];
optional-dependencies = { convert = [ accelerate numpy sentencepiece torch torchaudio torchvision transformers ]; docs = [ cairosvg mkdocs mkdocs-material mkdocstrings pillow ]; };
pythonImportsCheck = [ "ggml" ];
preCheck = '' rm -rf ggml '';
nativeCheckInputs = [ pytestCheckHook ];
passthru.updateScript = nix-update-script { extraArgs = [ "--version-regex" "^v([0-9.]+)$" ]; };
meta = { description = "Python bindings for ggml"; homepage = "https://github.com/abetlen/ggml-python"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ]; };})