diff options
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"; |