summaryrefslogtreecommitdiff
path: root/gnu/packages/machine-learning.scm
diff options
context:
space:
mode:
Diffstat (limited to 'gnu/packages/machine-learning.scm')
-rw-r--r--gnu/packages/machine-learning.scm10
1 files changed, 6 insertions, 4 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index 4d8ff4b5e3..061cb934a4 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -763,11 +763,13 @@ Performance is achieved by using the LLVM JIT compiler.")
(string-append (assoc-ref outputs "out")
"/bin")
"^test-")))))))
- (inputs (list curl glslang python python-gguf
- vulkan-headers vulkan-loader))
- (native-inputs (list pkg-config shaderc bash-minimal))
+ (inputs
+ (list curl glslang python-gguf python-minimal openblas spirv-headers
+ spirv-tools vulkan-headers vulkan-loader))
+ (native-inputs
+ (list bash-minimal pkg-config shaderc))
(propagated-inputs
- (list python-numpy python-pytorch python-sentencepiece openblas))
+ (list python-numpy python-pytorch python-sentencepiece))
(properties '((tunable? . #true))) ;use AVX512, FMA, etc. when available
(home-page "https://github.com/ggml-org/llama.cpp")
(synopsis "Port of Facebook's LLaMA model in C/C++")