{
  hardware.graphics.enable32Bit = true;
  hardware.nvidia-container-toolkit.enable = true;

  virtualisation.docker.enable = true;

  virtualisation.oci-containers = {
    backend = "docker";
    containers = {
      comfyui = {
        image = "ghcr.io/clsferguson/comfyui-docker:latest";
        autoStart = true;
        ports = [ "8188:8188" ];
        extraOptions = [
          "--device=nvidia.com/gpu=all"
          "--ipc=host"
        ];
        volumes = [
          "/home/jordan/ComfyUI/user:/app/ComfyUI/user"
          "/home/jordan/ComfyUI/custom_nodes:/app/ComfyUI/custom_nodes"
          "/home/jordan/ComfyUI/models:/app/ComfyUI/models:rw"
          "/home/jordan/ComfyUI/input:/app/ComfyUI/input:rw"
          "/home/jordan/ComfyUI/output:/app/ComfyUI/output:rw"
        ];
        environment = {
          TZ = "Europe/London";
          PUID = "1000";
          PGID = "1000";
          COMFY_AUTO_INSTALL = "1";
        };
      };
    };
  };
}