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