python312Packages.llama-cpp-python: use stdenv from cudaPackages
This commit is contained in:
@@ -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";
|
||||
|
||||
Reference in New Issue
Block a user