diff options
| author | Kirill Radzikhovskyy <kirillrdy@gmail.com> | 2024-10-27 04:45:16 +1100 |
|---|---|---|
| committer | Kirill Radzikhovskyy <kirillrdy@gmail.com> | 2024-11-03 08:05:38 +1100 |
| commit | f4e43ac273943bedb957d13fcdcf0cba4203e9bc (patch) | |
| tree | 3bd3e6115a8a47bc451ae16d9b8c489be14e2711 /pkgs/development/python-modules/llama-cpp-python/default.nix | |
| parent | cc1ac71df5506424d9a6f27969515bc930e08d74 (diff) | |
python312Packages.llama-cpp-python: use stdenv from cudaPackages
Diffstat (limited to 'pkgs/development/python-modules/llama-cpp-python/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/llama-cpp-python/default.nix | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/pkgs/development/python-modules/llama-cpp-python/default.nix b/pkgs/development/python-modules/llama-cpp-python/default.nix index 30eaf7ee6b77..4c0391858e4a 100644 --- a/pkgs/development/python-modules/llama-cpp-python/default.nix +++ b/pkgs/development/python-modules/llama-cpp-python/default.nix @@ -1,5 +1,6 @@ { lib, + stdenv, buildPythonPackage, cmake, fetchFromGitHub, @@ -23,14 +24,18 @@ scipy, huggingface-hub, }: - -buildPythonPackage rec { - pname = "llama-cpp-python"; +let version = "0.3.1"; +in +buildPythonPackage { + pname = "llama-cpp-python"; + inherit version; pyproject = true; disabled = pythonOlder "3.7"; + stdenv = if cudaSupport then cudaPackages.backendStdenv else stdenv; + src = fetchFromGitHub { owner = "abetlen"; repo = "llama-cpp-python"; |