Skip to content

Commit f4e43ac

Browse files
committed
python312Packages.llama-cpp-python: use stdenv from cudaPackages
1 parent cc1ac71 commit f4e43ac

File tree

1 file changed

+8
-3
lines changed
  • pkgs/development/python-modules/llama-cpp-python

1 file changed

+8
-3
lines changed

pkgs/development/python-modules/llama-cpp-python/default.nix

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
{
22
lib,
3+
stdenv,
34
buildPythonPackage,
45
cmake,
56
fetchFromGitHub,
@@ -23,14 +24,18 @@
2324
scipy,
2425
huggingface-hub,
2526
}:
26-
27-
buildPythonPackage rec {
28-
pname = "llama-cpp-python";
27+
let
2928
version = "0.3.1";
29+
in
30+
buildPythonPackage {
31+
pname = "llama-cpp-python";
32+
inherit version;
3033
pyproject = true;
3134

3235
disabled = pythonOlder "3.7";
3336

37+
stdenv = if cudaSupport then cudaPackages.backendStdenv else stdenv;
38+
3439
src = fetchFromGitHub {
3540
owner = "abetlen";
3641
repo = "llama-cpp-python";

0 commit comments

Comments
 (0)