diff options
author | Pol Dellaiera <pol.dellaiera@protonmail.com> | 2024-02-28 10:15:42 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-02-28 10:15:42 +0100 |
commit | 200390b3ed0bd1862767a11e3c3ce5a881dfdfc2 (patch) | |
tree | d9512148e01827b3bc283fd9d9e83accc31a4a28 /nixos/modules/services | |
parent | d852b9fdd7ccc3aa1607d5a51085babe42192207 (diff) | |
parent | 5c143f03663eb59a7a1eac4b24b7c034abc4f483 (diff) | |
download | nixlib-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.nix | 10 |
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 '' |