From 4a0ce7226f2334414d1c980c51baeb4e3bfd3519 Mon Sep 17 00:00:00 2001
From: Nick <nickjhiebert@proton.me>
Date: Fri, 18 Jul 2025 21:27:04 -0500
Subject: [PATCH] feat: added comfy ui

---
 modules/nixos/services/comfyui/default.nix | 22 +++++++++++++++-------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/modules/nixos/services/comfyui/default.nix b/modules/nixos/services/comfyui/default.nix
index 85afcdf..c965771 100755
--- a/modules/nixos/services/comfyui/default.nix
+++ b/modules/nixos/services/comfyui/default.nix
@@ -15,13 +15,21 @@ in
 
   services.comfyui = {
     enable = true;
-    # package = pkgs.comfyui-nvidia;
+    package = pkgs.comfyui-nvidia;
     host = localhost;
-    models = builtins.attrValues pkgs.nixified-ai.models;
-    customNodes = with comfyui.pkgs; [
-      comfyui-gguf
-      comfyui-impact-pack
-    ];
+    user = service.name;
+    acceleration = "cuda";
+    models = builtins.attrValues {
+      inherit (pkgs.nixified-ai)
+        models
+        ;
+    };
+    customNodes = builtins.attrValues {
+      inherit (comfyui.pkgs)
+        comfyui-gguf
+        comfyui-impact-pack
+        ;
+    };
   };
   fileSystems."/var/lib/${service.name}" = {
     device = service.paths.path0;
@@ -35,7 +43,7 @@ in
   };
 
   systemd.tmpfiles.rules = [
-    "Z ${service.paths.path0} 755 ${user0} users -"
+    "Z ${service.paths.path0} 755 ${service.name} ${service.name} -"
   ];
 
   networking = {