From f4e43ac273943bedb957d13fcdcf0cba4203e9bc Mon Sep 17 00:00:00 2001
From: Kirill Radzikhovskyy <kirillrdy@gmail.com>
Date: Sun, 27 Oct 2024 04:45:16 +1100
Subject: python312Packages.llama-cpp-python: use stdenv from cudaPackages

---
 pkgs/development/python-modules/llama-cpp-python/default.nix | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'pkgs/development/python-modules/llama-cpp-python')

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";
-- 
cgit v1.2.3