20 lines
443 B
Nix
20 lines
443 B
Nix
{
|
|
pkgs,
|
|
lib,
|
|
config,
|
|
...
|
|
}:
|
|
with lib; let
|
|
cfg = config.ollama;
|
|
in {
|
|
options.ollama = {enable = mkEnableOption "is ollama host";};
|
|
config = mkIf cfg.enable {
|
|
environment.systemPackages = [ pkgs.vscode pkgs.nodejs ];
|
|
services.ollama = {
|
|
enable = true;
|
|
package = pkgs.ollama-rocm;
|
|
# Optional: preload models, see https://ollama.com/library
|
|
loadModels = ["qwen2.5-coder" "deepseek-r1"];
|
|
};
|
|
};
|
|
}
|