python312Packages.llama-cpp-python: use stdenv from cudaPackages

This commit is contained in:
Kirill Radzikhovskyy
2024-11-03 08:05:38 +11:00
parent cc1ac71df5
commit f4e43ac273
@@ -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";