diff options
| author | Kirill Radzikhovskyy <kirillrdy@gmail.com> | 2024-10-26 06:50:47 +1100 |
|---|---|---|
| committer | Kirill Radzikhovskyy <kirillrdy@gmail.com> | 2024-11-03 08:05:38 +1100 |
| commit | cc1ac71df5506424d9a6f27969515bc930e08d74 (patch) | |
| tree | 3a26255e664885331487400b8b7e662113ac8e8b /pkgs/development/python-modules/llama-cpp-python | |
| parent | 17c58cde1623229a7f34973ca49fb9204badddda (diff) | |
python312Packages.llama-cpp-python: add passthru.test to build with CUDA
support
Diffstat (limited to 'pkgs/development/python-modules/llama-cpp-python')
| -rw-r--r-- | pkgs/development/python-modules/llama-cpp-python/default.nix | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/llama-cpp-python/default.nix b/pkgs/development/python-modules/llama-cpp-python/default.nix index 9a2f9458a875..30eaf7ee6b77 100644 --- a/pkgs/development/python-modules/llama-cpp-python/default.nix +++ b/pkgs/development/python-modules/llama-cpp-python/default.nix @@ -10,6 +10,7 @@ pytestCheckHook, pythonOlder, scikit-build-core, + llama-cpp-python, config, cudaSupport ? config.cudaSupport, @@ -86,6 +87,7 @@ buildPythonPackage rec { pythonImportsCheck = [ "llama_cpp" ]; passthru.updateScript = gitUpdater { rev-prefix = "v"; }; + passthru.tests.llama-cpp-python = llama-cpp-python.override { cudaSupport = true; }; meta = { description = "Python bindings for llama.cpp"; |