1{
 2  lib,
 3  llamaVersion,
 4  numpy,
 5  tqdm,
 6  requests,
 7  sentencepiece,
 8  pyyaml,
 9  poetry-core,
10  buildPythonPackage,
11  pytestCheckHook,
12}:
13
14buildPythonPackage {
15  pname = "gguf";
16  version = llamaVersion;
17  pyproject = true;
18  nativeBuildInputs = [ poetry-core ];
19  propagatedBuildInputs = [
20    numpy
21    tqdm
22    sentencepiece
23    pyyaml
24    requests
25  ];
26  src = lib.cleanSource ../../gguf-py;
27  pythonImportsCheck = [
28    "numpy"
29    "gguf"
30  ];
31  nativeCheckInputs = [ pytestCheckHook ];
32  doCheck = true;
33  meta = with lib; {
34    description = "Python package for writing binary files in the GGUF format";
35    license = licenses.mit;
36    maintainers = [ maintainers.ditsuke ];
37  };
38}