Something went wrong. Try again.
lol
Something went wrong. Try again.
1.6 kB · 100 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101{ lib, buildPythonPackage, fetchFromGitHub,
# 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.37"; pyproject = 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-QFpUGQ8m4c0SpHWnHhoyPdQkcywBToeLahDtG+JMcmA="; };
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 ];
meta = { description = "Python bindings for ggml"; homepage = "https://github.com/abetlen/ggml-python"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ GaetanLepage ]; };})