23 lines
596 B
Nix
23 lines
596 B
Nix
{ config, lib, inputs, ... }: {
|
|
|
|
options = {
|
|
sysconfig.services.ollama.enable = lib.options.mkOption {
|
|
type = lib.types.bool;
|
|
default = false;
|
|
};
|
|
};
|
|
|
|
config = lib.mkIf config.sysconfig.services.ollama.enable {
|
|
services.ollama = {
|
|
enable = true;
|
|
acceleration = "cuda";
|
|
package = let
|
|
pkgs-us = import inputs.nixpkgs-us {
|
|
system = "x86_64-linux";
|
|
config.allowUnfree = true;
|
|
};
|
|
in pkgs-us.ollama;
|
|
};
|
|
};
|
|
}
|