summaryrefslogtreecommitdiffstats
path: root/nixos/modules/services/misc
diff options
context:
space:
mode:
authorabysssol <abysssol@pm.me>2024-04-13 19:52:08 -0400
committerabysssol <abysssol@pm.me>2024-04-13 19:52:08 -0400
commit4508b5990092ce3d4760b6908667137c88adad5f (patch)
treecc4e41c97d38148556115c056532b2add87b822f /nixos/modules/services/misc
parent3769ffed5960fe7b9bf43383db9b295b9db902d1 (diff)
nixos/ollama: update documentation
Diffstat (limited to 'nixos/modules/services/misc')
-rw-r--r--nixos/modules/services/misc/ollama.nix11
1 files changed, 8 insertions, 3 deletions
diff --git a/nixos/modules/services/misc/ollama.nix b/nixos/modules/services/misc/ollama.nix
index 30c2b26d8322..b2c3de09015b 100644
--- a/nixos/modules/services/misc/ollama.nix
+++ b/nixos/modules/services/misc/ollama.nix
@@ -40,14 +40,19 @@ in
'';
};
acceleration = lib.mkOption {
- type = types.nullOr (types.enum [ "rocm" "cuda" ]);
+ type = types.nullOr (types.enum [ false "rocm" "cuda" ]);
default = null;
example = "rocm";
description = ''
What interface to use for hardware acceleration.
- - `rocm`: supported by modern AMD GPUs
- - `cuda`: supported by modern NVIDIA GPUs
+ - `null`: default behavior
+ if `nixpkgs.config.rocmSupport` is enabled, uses `"rocm"`
+ if `nixpkgs.config.cudaSupport` is enabled, uses `"cuda"`
+ otherwise defaults to `false`
+ - `false`: disable GPU, only use CPU
+ - `"rocm"`: supported by most modern AMD GPUs
+ - `"cuda"`: supported by most modern NVIDIA GPUs
'';
};
environmentVariables = lib.mkOption {