about summary refs log tree commit diff
path: root/nixos/modules/services
diff options
context:
space:
mode:
authorPol Dellaiera <pol.dellaiera@protonmail.com>2024-02-28 10:15:42 +0100
committerGitHub <noreply@github.com>2024-02-28 10:15:42 +0100
commit200390b3ed0bd1862767a11e3c3ce5a881dfdfc2 (patch)
treed9512148e01827b3bc283fd9d9e83accc31a4a28 /nixos/modules/services
parentd852b9fdd7ccc3aa1607d5a51085babe42192207 (diff)
parent5c143f03663eb59a7a1eac4b24b7c034abc4f483 (diff)
downloadnixlib-200390b3ed0bd1862767a11e3c3ce5a881dfdfc2.tar
nixlib-200390b3ed0bd1862767a11e3c3ce5a881dfdfc2.tar.gz
nixlib-200390b3ed0bd1862767a11e3c3ce5a881dfdfc2.tar.bz2
nixlib-200390b3ed0bd1862767a11e3c3ce5a881dfdfc2.tar.lz
nixlib-200390b3ed0bd1862767a11e3c3ce5a881dfdfc2.tar.xz
nixlib-200390b3ed0bd1862767a11e3c3ce5a881dfdfc2.tar.zst
nixlib-200390b3ed0bd1862767a11e3c3ce5a881dfdfc2.zip
Merge pull request #291748 from abysssol/update-ollama-0.1.27
ollama: 0.1.26 -> 0.1.27
Diffstat (limited to 'nixos/modules/services')
-rw-r--r--nixos/modules/services/misc/ollama.nix10
1 files changed, 6 insertions, 4 deletions
diff --git a/nixos/modules/services/misc/ollama.nix b/nixos/modules/services/misc/ollama.nix
index 0d3574a2bac0..7784f3170cef 100644
--- a/nixos/modules/services/misc/ollama.nix
+++ b/nixos/modules/services/misc/ollama.nix
@@ -1,11 +1,13 @@
 { config, lib, pkgs, ... }:
 let
-  inherit (lib.types) nullOr enum;
+  inherit (lib) types;
 
   cfg = config.services.ollama;
   ollamaPackage = cfg.package.override {
     inherit (cfg) acceleration;
-    linuxPackages.nvidia_x11 = config.hardware.nvidia.package;
+    linuxPackages = config.boot.kernelPackages.overrideAttrs {
+      nvidia_x11 = config.hardware.nvidia.package;
+    };
   };
 in
 {
@@ -15,14 +17,14 @@ in
         lib.mdDoc "Server for local large language models"
       );
       listenAddress = lib.mkOption {
-        type = lib.types.str;
+        type = types.str;
         default = "127.0.0.1:11434";
         description = lib.mdDoc ''
           Specifies the bind address on which the ollama server HTTP interface listens.
         '';
       };
       acceleration = lib.mkOption {
-        type = nullOr (enum [ "rocm" "cuda" ]);
+        type = types.nullOr (types.enum [ "rocm" "cuda" ]);
         default = null;
         example = "rocm";
         description = lib.mdDoc ''