From 8db387e8a54cc87adf78eddd9f9d9a24b079e6fb Mon Sep 17 00:00:00 2001
From: Mirko Lenz <mirko@mirkolenz.com>
Date: Wed, 4 Dec 2024 15:12:14 +0100
Subject: [PATCH] fix: selectively use cuda

---
 flake-modules/default.nix | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/flake-modules/default.nix b/flake-modules/default.nix
index fc8f575..3267461 100644
--- a/flake-modules/default.nix
+++ b/flake-modules/default.nix
@@ -10,7 +10,7 @@ let
     inherit system;
     config = {
       allowUnfree = true;
-      cudaSupport = true;
+      cudaSupport = false;
     };
     overlays = lib.singleton (
       final: prev: {
@@ -18,6 +18,9 @@ let
           enableNvidiaContainerCli = false;
           forceNvcCli = false;
         };
+        ollama = final.unstable.ollama.override {
+          acceleration = "cuda";
+        };
         system-manager = inputs.system-manager.packages.${system}.default;
         nixglhost = inputs.nixglhost.packages.${system}.default;
         stable = prev;
@@ -25,10 +28,10 @@ let
           inherit system;
           config = {
             allowUnfree = true;
-            cudaSupport = true;
+            cudaSupport = false;
           };
         };
-        inherit (final.unstable) uv;
+        inherit (final.unstable) uv open-webui;
       }
     );
   };