diff --git a/hosts/workstation/ollama.nix b/hosts/workstation/ollama.nix new file mode 100644 index 0000000..2afe3af --- /dev/null +++ b/hosts/workstation/ollama.nix @@ -0,0 +1,19 @@ +{ + pkgs, + lib, + config, + ... +}: +with lib; let + cfg = config.ollama; +in { + options.ollama = {enable = mkEnableOption "is ollama host";}; + config = mkIf cfg.enable { + services.ollama = { + enable = true; + package = pkgs.ollama-rocm; + # Optional: preload models, see https://ollama.com/library + loadModels = ["qwen2.5-coder:1.5b"]; + }; + }; +}