{ 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";
    };
  };
}