nix-conf/modules/linux/nixos/ollama.nix

22 lines
494 B
Nix

{
config,
pkgs,
lib,
...
}: {
options = {
crony.ollama.enable = lib.mkEnableOption "Enable and setup ollama";
};
config = lib.mkIf config.crony.ollama.enable {
services.ollama = {
enable = true;
package = pkgs.ollama-cuda;
loadModels = ["gpt-oss:20b"];
environmentVariables = {OLLAMA_KEEP_ALIVE = "2m";};
};
# systemd.services.ollama.wantedBy = lib.mkForce [];
# systemd.services.ollama-model-loader.wantedBy = lib.mkForce [];
};
}