comfyui

.gitignore

@@ -1 +1,3 @@
 /.direnv/
+/venv-comfyui/
+/ComfyUI/

flake.nix

@@ -17,9 +17,13 @@
           };
         in
         basecamp.mkShell pkgs {
+          env.LD_LIBRARY_PATH = "${pkgs.stdenv.cc.cc.lib}/lib";
           packages = [
-            (pkgs.python3.withPackages (_python-pkgs: [
-              # python-pkgs.drawsvg
+            pkgs.just
+
+            pkgs.gcc # For libstdc++.so.6
+
+            (pkgs.python312.withPackages (_python-pkgs: [
               # python-pkgs.ruff
               # python-pkgs.jedi-language-server
             ]))

justfile

@@ -12,3 +12,18 @@ fmt:
 # Perform automated fixes
 fix:
     bc-fix
+
+# Install ComfyUI
+setup:
+    #!/usr/bin/env bash
+    python -m venv venv-comfyui
+    source venv-comfyui/bin/activate
+    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu128
+    git clone git@github.com:comfyanonymous/ComfyUI.git
+    pip install -r ComfyUI/requirements.txt
+
+# Run ComfyUI
+run:
+    #!/usr/bin/env bash
+    source venv-comfyui/bin/activate
+    python ComfyUI/main.py