18 lines
351 B
Nix
18 lines
351 B
Nix
{ config, pkgs, ... }:
|
|
{
|
|
services.ollama = {
|
|
enable = true;
|
|
package = pkgs.callPackage ./custom/ollama.nix {
|
|
acceleration = "rocm";
|
|
config = {
|
|
rocmSupport = true;
|
|
cudaSupport = false;
|
|
};
|
|
};
|
|
acceleration = "rocm";
|
|
environmentVariables = {
|
|
HSA_OVERRIDE_GFX_VERSION = "10.3.0";
|
|
};
|
|
};
|
|
}
|